ezKit 1.0.0__py3-none-any.whl

ezKit/weixin.py

@@ -0,0 +1,148 @@
+import json
+import time
+
+import requests
+
+
+class WeiXin(object):
+
+    '''
+    企业微信开发者中心
+
+        https://developer.work.weixin.qq.com/
+        https://developer.work.weixin.qq.com/document/path/90313 (全局错误码)
+
+    参考文档:
+
+        https://www.gaoyuanqi.cn/python-yingyong-qiyewx/
+        https://www.jianshu.com/p/020709b130d3
+    '''
+
+    _work_id, _agent_id, _agent_secret, _access_token = None, None, None, None
+
+    def __init__(self, work_id, agent_id, agent_secret):
+        ''' Initiation '''
+        self._work_id = work_id
+        self._agent_id = agent_id
+        self._agent_secret = agent_secret
+
+        ''' 获取 Token '''
+        self.get_access_token()
+
+    def get_access_token(self):
+        _response = requests.get(f'https://qyapi.weixin.qq.com/cgi-bin/gettoken?corpid={self._work_id}&corpsecret={self._agent_secret}')
+        if _response.status_code == 200:
+            _result = _response.json()
+            self._access_token = _result.get('access_token')
+        else:
+            self._access_token = None
+
+    def get_agent_list(self):
+
+        self.get_access_token() if self._access_token == None else next
+
+        _response = requests.get(f'https://qyapi.weixin.qq.com/cgi-bin/agent/list?access_token={self._access_token}')
+
+        if _response.status_code == 200:
+            _response_data = _response.json()
+            if _response_data.get('errcode') == 42001:
+                self.get_access_token()
+                time.sleep(1)
+                self.get_agent_list()
+            return _response_data
+        return {'response': _response.text}
+
+    def get_department_list(self, id):
+
+        self.get_access_token() if self._access_token == None else next
+
+        _response = requests.get(f'https://qyapi.weixin.qq.com/cgi-bin/department/list?access_token={self._access_token}&id={id}')
+
+        if _response.status_code == 200:
+            _response_data = _response.json()
+            if _response_data.get('errcode') == 42001:
+                self.get_access_token()
+                time.sleep(1)
+                self.get_department_list(id)
+            return _response_data
+        return {'response': _response.text}
+
+    def get_user_list(self, id):
+
+        self.get_access_token() if self._access_token == None else next
+
+        _response = requests.get(f'https://qyapi.weixin.qq.com/cgi-bin/user/list?access_token={self._access_token}&department_id={id}')
+
+        if _response.status_code == 200:
+            _response_data = _response.json()
+            if _response_data.get('errcode') == 42001:
+                self.get_access_token()
+                time.sleep(1)
+                self.get_user_list(id)
+            return _response_data
+        return {'response': _response.text}
+
+    def get_user_id_by_mobile(self, mobile):
+
+        self.get_access_token() if self._access_token == None else next
+
+        _json_dict = {'mobile': mobile}
+
+        _json_string = json.dumps(_json_dict)
+
+        _response = requests.post(f'https://qyapi.weixin.qq.com/cgi-bin/user/getuserid?access_token={self._access_token}', data=_json_string)
+
+        if _response.status_code == 200:
+            _response_data = _response.json()
+            if _response_data.get('errcode') == 42001:
+                self.get_access_token()
+                time.sleep(1)
+                self.get_user_id_by_mobile(id)
+            return _response_data
+        return {'response': _response.text}
+
+    def get_user_info(self, id):
+
+        self.get_access_token() if self._access_token == None else next
+
+        _response = requests.get(f'https://qyapi.weixin.qq.com/cgi-bin/user/get?access_token={self._access_token}&userid={id}')
+
+        if _response.status_code == 200:
+            _response_data = _response.json()
+            if _response_data.get('errcode') == 42001:
+                self.get_access_token()
+                time.sleep(1)
+                self.get_user_info(id)
+            return _response_data
+        return {'response': _response.text}
+
+    def send_text(self, users, message):
+        '''
+        https://developer.work.weixin.qq.com/document/path/90235
+        '''
+
+        self.get_access_token() if self._access_token == None else next
+
+        _json_dict = {
+            'touser': users,
+            'msgtype': 'text',
+            'agentid': self._agent_id,
+            'text': {'content': message},
+            'safe': 0,
+            'enable_id_trans': 0,
+            'enable_duplicate_check': 0,
+            'duplicate_check_interval': 1800
+        }
+
+        _json_string = json.dumps(_json_dict)
+
+        _response = requests.post(f'https://qyapi.weixin.qq.com/cgi-bin/message/send?access_token={self._access_token}', data=_json_string)
+
+        if _response.status_code == 200:
+            _response_data = _response.json()
+            if _response_data.get('errcode') == 42001:
+                self.get_access_token()
+                time.sleep(1)
+                self.send_text(users, message)
+            return _response_data
+        return {'response': _response.text}

ezKit/xftp.py

@@ -0,0 +1,194 @@
+'''
+ftplib: https://docs.python.org/3.10/library/ftplib.html
+'''
+import os
+from ftplib import FTP
+from pathlib import Path
+
+
+class XFTP:
+
+    def __init__(self, host='127.0.0.1', port=21, username='anonymous', password='', encoding='UTF-8', debuglevel=0):
+        ''' Initiation '''
+        self.ftp = FTP()
+        self.ftp.set_debuglevel(debuglevel)
+        self.host = host
+        self.port = port
+        self.username = username
+        self.password = password
+        self.encoding = encoding
+        self.retry = 1
+
+    def connect(self):
+        ''' FTP connect '''
+        try:
+            self.ftp.connect(host=self.host, port=self.port, timeout=10)
+            self.ftp.encoding = self.encoding
+            self.ftp.login(user=self.username, passwd=self.password)
+            print('FTP connect success')
+            print('-' * 80)
+            return True
+        except Exception as e:
+            print(f'FTP connect error: {e}, retry...')
+            if self.retry >= 3:
+                print('FTP connect faild')
+                return False
+            self.retry += 1
+            self.connect()
+
+    def close(self, info=None):
+        ''' FTP close '''
+        print(info) if info else None
+        try:
+            self.ftp.quit()
+        except:
+            self.ftp.close()
+        print('-' * 80)
+        print('FTP connect closed')
+
+    def get_file_list(self, dir='/'):
+        ''' Get file list '''
+        self.chdir_to_remote(dir)
+        return self.ftp.nlst()
+
+    def get_file_size(self, dir='/', file=None):
+        ''' Get file size '''
+        self.chdir_to_remote(dir)
+        return self.ftp.size(file)
+
+    def mkdir(self, dir_string='/'):
+        ''' 创建目录 (从 / 目录依次递增创建子目录. 如果目录存在, 创建目录时会报错, 所以这里忽略所有错误.) '''
+        try:
+            dir_list = dir_string.split("/")
+            for i, _ in enumerate(dir_list):
+                dir = '/'.join(dir_list[:i + 1])
+                try:
+                    self.ftp.mkd(dir)
+                except:
+                    pass
+            return True
+        except:
+            return False
+
+    def chdir_to_remote(self, dir='/'):
+        ''' change to remote directory'''
+        try:
+            self.ftp.cwd(dir)
+        except:
+            self.close(f'Remote directory error: {dir}')
+
+    def x_exit(self, info=None):
+        ''' Exit '''
+        print(info) if info else None
+        # 注意: exit() 并不会退出脚本, 配合 try 使用
+        exit()
+
+    def x_exec(self, local_dir='.', local_file='', remote_dir='/', remote_file='', upload=False):
+        ''' Download or Upload '''
+
+        bufsize = 1024
+        local_path = local_dir + '/' + local_file
+        remote_path = remote_dir + '/' + remote_file
+        info = 'Upload' if upload else 'Download'
+
+        # 检查参数
+        if upload:
+            if local_file == '':
+                self.close('Argument Miss: local file')
+            # 如果没有设置 远程文件 名称, 则使用 本地文件 名称
+            if remote_file == '':
+                remote_file = local_file
+                remote_path = remote_dir + '/' + remote_file
+        else:
+            if remote_file == '':
+                self.close('Argument Miss: remote file')
+            # 如果没有设置 本地文件 名称, 则使用 远程文件 名称
+            if local_file == '':
+                local_file = remote_file
+                local_path = local_dir + '/' + local_file
+
+        # 进入本地目录
+        try:
+            if upload:
+                # 检查本地目录
+                stat = Path(local_dir)
+                self.close(f'Local directory error: {local_dir}') if stat.exists() == False else None
+            else:
+                # 创建本地目录
+                Path(local_dir).mkdir(parents=True, exist_ok=True)
+            # 进入本地目录
+            os.chdir(local_dir)
+        except:
+            # 第一层 try 使用 self.x_exit() 无效, 直接使用 self.close()
+            self.close(f'Local directory error: {local_dir}')
+
+        # 上传或下载
+        try:
+
+            if upload:
+
+                ''' 上传 '''
+
+                # 创建远程目录
+                if remote_dir != '/':
+                    self.mkdir(remote_dir)
+
+                # 进入远程目录
+                self.chdir_to_remote(remote_dir)
+
+                # 上传文件
+                stat = Path(local_file)
+                if stat.exists() and stat.is_file():
+                    with open(local_file, 'rb') as fid:
+                        self.ftp.storbinary(f'STOR {remote_file}', fid, bufsize)
+                    print('{} success: {} -> {}'.format(info, local_path.replace('//', '/'), remote_path.replace('//', '/')))
+                    return True
+                else:
+                    self.x_exit('{} error: {} is not exist'.format(info, local_path.replace('//', '/')))
+
+            else:
+
+                ''' 下载 '''
+
+                # 进入远程目录
+                self.chdir_to_remote(remote_dir)
+
+                # 下载文件
+                if remote_file in self.ftp.nlst():
+                    with open(local_file, 'wb') as fid:
+                        self.ftp.retrbinary(f'RETR {remote_file}', fid.write, bufsize)
+                    print('{} success: {} -> {}'.format(info, remote_path.replace('//', '/'), local_path.replace('//', '/')))
+                    return True
+                else:
+                    self.x_exit('{} error: {} is not exist'.format(info, remote_path.replace('//', '/')))
+
+        except Exception as e:
+            # 第一层 try 使用 self.x_exit() 无效, 直接使用 self.close()
+            # self.close('{} faild! Please check {} or {}'.format(info, local_path, remote_path))
+            self.close(f'{info} error: {e}')
+            return False
+
+    def handle_all(self, local_dir='.', remote_dir='/', upload=False):
+        ''' Handle All '''
+        if upload:
+            # 检查本地目录
+            stat = Path(local_dir)
+            self.close(f'Local directory error: {local_dir}') if stat.exists() == False else None
+            # 获取文件列表
+            local_files = [f for f in os.listdir(local_dir) if os.path.isfile(os.path.join(local_dir, f))]
+            for i in local_files:
+                self.x_exec(local_dir=local_dir, remote_dir=remote_dir, local_file=i, upload=True)
+        else:
+            remote_files = self.get_file_list(remote_dir)
+            for i in remote_files:
+                self.x_exec(local_dir=local_dir, remote_dir=remote_dir, remote_file=i)
+
+    def retrlines(self, remote_dir='/', cmd='LIST'):
+        ''' Retrlines '''
+        try:
+            self.chdir_to_remote(remote_dir)
+            print(self.ftp.retrlines(cmd))
+            self.close()
+        except Exception as e:
+            # 第一层 try 使用 self.x_exit() 无效, 直接使用 self.close()
+            self.close(e)