ifkit 0.1.51__tar.gz

ifkit-0.1.51/LICENSE

@@ -0,0 +1,7 @@
+Copyright 2025 infuq
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

ifkit-0.1.51/MANIFEST.in

@@ -0,0 +1 @@
+include README.md LICENSE

ifkit-0.1.51/PKG-INFO

@@ -0,0 +1,23 @@
+Metadata-Version: 2.1
+Name: ifkit
+Version: 0.1.51
+Summary: 日常工作使用的小工具/小功能
+Home-page: https://www.infuq.com
+Author: infuq
+Author-email: 864511259@qq.com
+License: MIT
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.6
+Classifier: Programming Language :: Python :: Implementation :: CPython
+Classifier: Programming Language :: Python :: Implementation :: PyPy
+Requires-Python: >=3.6.0
+Description-Content-Type: text/markdown
+License-File: LICENSE
+
+
+日常工作使用的小工具/小功能
+
+
+[个人站点](https://www.infuq.com)

ifkit-0.1.51/README.md

@@ -0,0 +1,4 @@
+日常工作使用的小工具/小功能
+
+
+[个人站点](https://www.infuq.com)

ifkit-0.1.51/ifkit/__init__.py

@@ -0,0 +1,9 @@
+from .csv import *
+from .excel import ReadExcelUtil, WriteXlsxUtil
+from .json import *
+from .mq import *
+from .qyweixin import *
+from .sms import *
+from .txt import *
+from .mysql_export_excel import export_excel
+

ifkit-0.1.51/ifkit/__version__.py

@@ -0,0 +1,8 @@
+# 8b    d8 Yb  dP 88""Yb    db     dP""b8 88  dP    db     dP""b8 888888
+# 88b  d88  YbdP  88__dP   dPYb   dP   `" 88odP    dPYb   dP   `" 88__
+# 88YbdP88   8P   88"""   dP__Yb  Yb      88"Yb   dP__Yb  Yb  "88 88""
+# 88 YY 88  dP    88     dP""""Yb  YboodP 88  Yb dP""""Yb  YboodP 888888
+
+VERSION = (5, 2, 0)
+
+__version__ = '.'.join(map(str, VERSION))

ifkit-0.1.51/ifkit/csv.py

@@ -0,0 +1,18 @@
+import csv
+from collections.abc import Iterable
+from collections.abc import Iterator
+
+# 读取 CSV 文件
+def read_csv(file_name, encoding='UTF8', start_read_row_num=0):
+    row_num = 0
+    with open(file_name, 'r', encoding=encoding) as f: # encoding='UTF8'   encoding='GBK'
+        reader = csv.DictReader(f)
+        for row_data in reader:
+            if row_num < start_read_row_num:
+                row_num = row_num + 1
+                continue
+
+            rn = row_num
+            row_num = row_num + 1
+            yield (rn, row_data)
+

ifkit-0.1.51/ifkit/excel.py

@@ -0,0 +1,173 @@
+"""
+读取xlsx或xls文件
+
+pip install xlrd
+pip install xlwt
+pip install openpyxl
+"""
+
+import openpyxl
+import os
+import xlrd
+import xlwt
+from collections.abc import Iterable
+from collections.abc import Iterator
+import datetime
+import re
+
+
+# 读取文件
+class ReadExcelUtil(Iterable):
+
+    def __init__(self, file_name, start_read_row_num=0):
+        self.filename = file_name
+        self.suffix = os.path.splitext(self.filename)[-1][1:].lower() # xls xlsx
+        self.start_read_row_num = start_read_row_num # 开始从第N行读取数据
+            
+        
+        if self.suffix == 'xlsx':
+            self.workbook = openpyxl.load_workbook(self.filename, data_only=True)
+            self.sheetnames = self.workbook.sheetnames # Sheet1 Sheet2 Sheet3...
+            self.sheet = self.workbook[self.sheetnames[0]] # 根据名称默认读取第一个Sheet
+
+        if self.suffix == 'xls':
+            self.workbook = xlrd.open_workbook(self.filename, encoding_override="UTF-8")
+            self.sheetnames = self.workbook._sheet_names # Sheet1 Sheet2 Sheet3...
+            # self.sheet = self.workbook[0] # 根据下标默认读取第一个Sheet
+            self.sheet = self.workbook[self.sheetnames[0]] # 根据名称默认读取第一个Sheet
+
+    def __iter__(self):
+        if self.suffix == 'xlsx':
+            return ReadXlsxIterator(self.sheet, self.start_read_row_num)
+        if self.suffix == 'xls':
+            return ReadXlsIterator(self.sheet, self.start_read_row_num)
+
+
+# 迭代器 xlsx
+class ReadXlsxIterator(Iterator):
+
+    def __init__(self, sheet, start_read_row_num):
+        self.sheet = sheet
+        self.max_row = sheet.max_row
+        self.max_column = sheet.max_column
+        self.row_num = start_read_row_num
+
+    def __next__(self):
+        try:
+            if self.row_num <= self.max_row:
+                # 读取行内容
+                row_data = []
+                for i in range(1, self.max_column + 1):
+                    cell_value = self.sheet.cell(row=self.row_num, column=i).value
+                    row_data.append(cell_value)
+                
+                rn = self.row_num
+                self.row_num += 1
+                # (行号, 行内容)
+                return (rn, row_data)
+        except Exception:
+            raise StopIteration
+        raise StopIteration
+
+
+# 迭代器 xls
+class ReadXlsIterator(Iterator):
+
+    def __init__(self, sheet, start_read_row_num):
+        self.sheet = sheet
+        self.nrows = sheet.nrows
+        self.row_num = start_read_row_num
+
+    def __next__(self):
+        try:
+            if self.row_num <= self.nrows:
+                # 读取行内容
+                # (行号, 行内容)
+                rn = self.row_num
+                self.row_num += 1
+                return (rn, self.sheet.row_values(rn))
+        except Exception:
+            raise StopIteration
+        raise StopIteration
+
+
+# 写入文件
+class WriteXlsxUtil(object):
+
+    def __init__(self, file_name):
+        self.index = 0
+        self.suffix = '.xlsx'
+        self.workbook = openpyxl.Workbook()
+        self.sheet = self.workbook.create_sheet(index=self.index, title="Sheet")
+        self.empty = True  # 标记是否真的写入了内容
+        self.row = 0
+        if file_name is None:
+            self.file_name = "File-" + datetime.datetime.now().strftime('%Y-%m-%d %H-%M-%S') + self.suffix
+        else:
+            self.file_name = file_name + self.suffix
+
+    
+    def set_sheet(self, sheet_name):
+        self.index = self.index + 1
+        self.sheet = self.workbook.create_sheet(index=self.index, title=sheet_name)
+        self.row = 0
+    
+    # 设置列宽
+    def column_width(self, index, width):
+        self.sheet.column_dimensions[index].width = width    
+
+    # 写入单元格内容
+    def write(self, content):
+        self.empty = False
+        self.row += 1
+        for index in range(1, len(content) + 1):  # index = column
+            if isinstance(content[index - 1], str) or isinstance(content[index - 1], (int, float)):
+                v = str(content[index - 1])
+                digit = re.match(r'^\d+(\.\d+)?$', v, re.I)
+                if digit:
+                    self.sheet.cell(self.row, index).value = v
+                else:
+                    self.sheet.cell(self.row, index).value = v
+            else:
+                cell = content[index - 1]
+                self.sheet.cell(self.row, index).value = cell['value']
+                if 'color' in cell:
+                    self.sheet.cell(self.row, index).font = openpyxl.styles.Font(color=cell['color'])
+
+    # 保存文件
+    def save(self):
+        if not self.empty:
+            self.workbook.save("./" + self.file_name)
+
+
+
+# class WriteXlsUtil(object):
+#
+#     def __init__(self, file_name='File'):
+#         self.suffix = '.xls'
+#         self.workbook = xlwt.Workbook()
+#         self.sheet = self.workbook.add_sheet("Sheet")
+#         self.file_name = file_name
+#         self.empty = True
+#         self.row = -1
+#
+#     # 写入单元格内容
+#     def write(self, content):
+#         self.empty = False
+#         self.row += 1
+#         for index in range(len(content)):
+#             cnt = str(content[index])
+#             if cnt.isdigit():
+#                 self.sheet.write(self.row, index, cnt.zfill(len(cnt)))
+#             else:
+#                 self.sheet.write(self.row, index, content[index])
+#
+#     # 保存文件
+#     def save(self):
+#         if not self.empty:
+#             self.workbook.save("./" + self.file_name + "-" + datetime.datetime.now().strftime('%Y-%m-%d %H-%M-%S') + self.suffix)
+
+
+
+
+__all__ = ['ReadExcelUtil', 'WriteXlsxUtil']

ifkit-0.1.51/ifkit/json.py

@@ -0,0 +1,11 @@
+from datetime import date, datetime
+import json
+
+class ComplexEncoder(json.JSONEncoder):
+    def default(self, obj):
+        if isinstance(obj, datetime):
+            return obj.strftime('%Y-%m-%d %H:%M:%S')
+        elif isinstance(obj, date):
+            return obj.strftime('%Y-%m-%d')
+        else:
+            return json.JSONEncoder.default(self, obj)

ifkit-0.1.51/ifkit/mq.py

@@ -0,0 +1,62 @@
+"""
+pip install mq_http_sdk
+"""
+
+import sys
+import time
+from mq_http_sdk.mq_exception import MQExceptionBase
+from mq_http_sdk.mq_producer import *
+from mq_http_sdk.mq_client import *
+
+
+# 生产者通过HTTP协议发送消息
+def send_http(host, access_id, access_key, instance_id, topic_name, message_body, message_tag, message_key):
+    mq_client = MQClient(host=host, access_id=access_id, access_key=access_key)
+
+    producer = mq_client.get_producer(instance_id, topic_name)
+    
+    try:
+        msg = TopicMessage(message_body, message_tag)           
+        msg.put_property("__key", "__value")  # 设置属性
+        msg.set_message_key(message_key)
+        # msg.set_sharding_key("")
+
+        re_msg = producer.publish_message(msg)        
+        return re_msg.message_id
+    except MQExceptionBase as e:
+        if e.type == "TopicNotExist":
+            print("Topic not exist, please create it.")            
+        print("Publish Message Fail. Exception:%s" % e)
+
+    return None
+
+
+# 消费者通过HTTP协议消费消息
+def consume_loop_http(host, access_id, access_key, instance_id, topic_name, group_id, callback):
+
+    mq_client = MQClient(host=host, access_id=access_id, access_key=access_key)    
+    consumer = mq_client.get_consumer(instance_id, topic_name, group_id)
+    wait_seconds = 3
+    batch = 3
+    while True:
+        try:
+            # recv_msgs = consumer.consume_message_orderly(batch, wait_seconds)
+            recv_msgs = consumer.consume_message(batch, wait_seconds)
+            for message in recv_msgs:
+                # 回调业务方法
+                callback(message)
+                consumer.ack_message(message.receipt_handle.split())                
+                
+        except MQExceptionBase as e:
+            if e.type == "MessageNotExist":
+                # print(("No new message! RequestId: %s" % e.req_id))                
+                # continue
+                # break
+                time.sleep(5)
+                continue
+
+            print(("Consume Message Fail! Exception:%s\n" % e))
+            time.sleep(2)
+            continue
+
+

ifkit-0.1.51/ifkit/mysql_all_tables.py

@@ -0,0 +1,31 @@
+#! /usr/bin/env python
+
+import pymysql
+
+
+def _find_all_tables(cursor, table_schema):
+    sql = "SELECT * FROM information_schema.tables WHERE table_schema = '%s';"
+    sql = sql % (table_schema,)
+    cursor.execute(sql)
+    all_data = cursor.fetchall()
+    all_tables = []
+    for data in all_data:
+        all_tables.append(data['TABLE_NAME'])
+    return all_tables
+
+
+def find_all_tables(host, port, user, passwd, db):
+    conn = pymysql.connect(host=host,
+                           port=port,
+                           user=user,
+                           passwd=passwd,
+                           db=db)
+    cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)
+
+    all_tables = _find_all_tables(cursor, db)
+
+    cursor.close()
+    conn.close()
+
+    return all_tables
+

ifkit-0.1.51/ifkit/mysql_create_table_statement.py

@@ -0,0 +1,14 @@
+#! /usr/bin/env python
+
+def show_create_table(cursor, table_name):
+    try:
+        sql = "show create table %s"
+        sql = sql % (table_name,)
+        cursor.execute(sql)
+        all_data = cursor.fetchall()
+        return all_data[0]['Create Table']
+    except Exception as e:
+        print(e)
+
+
+

ifkit-0.1.51/ifkit/mysql_export_excel.py

@@ -0,0 +1,63 @@
+#! /usr/bin/env python
+
+import decimal
+import re
+
+import pymysql
+
+
+def export_excel(host, port, user, passwd, db, query_sql, file_name=None):
+    conn = pymysql.connect(host=host,
+                           port=port,
+                           user=user,
+                           passwd=passwd,
+                           db=db)
+    cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)
+
+    from .excel import WriteXlsxUtil
+    write_book = WriteXlsxUtil(file_name)
+
+    cursor.execute(query_sql)
+    alias_field = alias(cursor.description)
+    write_book.write(tuple(alias_field))
+
+    all_data = cursor.fetchall()
+    for data in all_data:
+        data_list = []
+        for field in alias_field:
+            value = data[field]
+            data_list.append(replace(value))
+        write_book.write(tuple(data_list))
+
+    # 保存文件
+    write_book.save()
+
+    # 关闭数据库连接
+    cursor.close()
+    conn.close()
+
+
+def alias(description):
+    alias_field = []
+    for i in range(len(description)):
+        alias_field.append(description[i][0])
+    return alias_field
+
+def replace(data):
+    if data:
+        if isinstance(data, decimal.Decimal):
+            if data == 0:
+                return 0
+            else:
+                return replace(str(data))
+        elif isinstance(data, str):
+            # 70.10 -> 70.1
+            data = re.sub(r'^(.*\.\d+?)0+$', r'\1', data)
+            # 70.00 -> 70
+            data = re.sub(r'^(.*)\.0+$', r'\1', data)
+
+            return data
+        else:
+            return replace(str(data))
+    else:
+        return ''

ifkit-0.1.51/ifkit/qyweixin.py

@@ -0,0 +1,13 @@
+import requests
+
+def send_markdown_message(content, key):
+    url = 'https://qyapi.weixin.qq.com/cgi-bin/webhook/send?key=%s'
+    url = url % (key,)
+    payload = {
+        'msgtype': 'markdown',
+        'markdown': {
+            'content': content
+        }
+    }
+    response = requests.post(url, json=payload)
+    return response.json()

ifkit-0.1.51/ifkit/sms.py

@@ -0,0 +1,21 @@
+import requests
+
+# 发送短信
+def send_sms(content, accesskey, secret, sign, templateId, mobile):
+
+    headers = {
+        'User-Agent': 'Mozilla/5.0 (Windows NT 6.1; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/79.0.3945.88 Safari/537.36'
+    }
+
+    data = {
+        'accesskey': accesskey,
+        'secret': secret,
+        'sign': sign,
+        'templateId': templateId,
+        'mobile': mobile,
+        'content': content
+    }
+
+    # 发送
+    response = requests.post('http://api.1cloudsp.com/api/v2/single_send', data=data, headers=headers)
+    return response.text

ifkit-0.1.51/ifkit/txt.py

@@ -0,0 +1,13 @@
+
+# 读取 txt 文件
+def read_txt(file_name, encoding='GBK', start_read_row_num=0):
+    row_num = 0
+    with open(file_name, 'r', encoding=encoding) as f:
+        for row_data in f.readlines():
+            if row_num < start_read_row_num:
+                row_num = row_num + 1
+                continue
+
+            rn = row_num
+            row_num = row_num + 1
+            yield (rn, row_data)

ifkit-0.1.51/ifkit.egg-info/PKG-INFO

@@ -0,0 +1,23 @@
+Metadata-Version: 2.1
+Name: ifkit
+Version: 0.1.51
+Summary: 日常工作使用的小工具/小功能
+Home-page: https://www.infuq.com
+Author: infuq
+Author-email: 864511259@qq.com
+License: MIT
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.6
+Classifier: Programming Language :: Python :: Implementation :: CPython
+Classifier: Programming Language :: Python :: Implementation :: PyPy
+Requires-Python: >=3.6.0
+Description-Content-Type: text/markdown
+License-File: LICENSE
+
+
+日常工作使用的小工具/小功能
+
+
+[个人站点](https://www.infuq.com)

ifkit-0.1.51/ifkit.egg-info/SOURCES.txt

@@ -0,0 +1,20 @@
+LICENSE
+MANIFEST.in
+README.md
+setup.py
+ifkit/__init__.py
+ifkit/__version__.py
+ifkit/csv.py
+ifkit/excel.py
+ifkit/json.py
+ifkit/mq.py
+ifkit/mysql_all_tables.py
+ifkit/mysql_create_table_statement.py
+ifkit/mysql_export_excel.py
+ifkit/qyweixin.py
+ifkit/sms.py
+ifkit/txt.py
+ifkit.egg-info/PKG-INFO
+ifkit.egg-info/SOURCES.txt
+ifkit.egg-info/dependency_links.txt
+ifkit.egg-info/top_level.txt

ifkit-0.1.51/ifkit.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

ifkit-0.1.51/ifkit.egg-info/top_level.txt

@@ -0,0 +1 @@
+ifkit

ifkit-0.1.51/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

ifkit-0.1.51/setup.py

@@ -0,0 +1,131 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+# Note: To use the 'upload' functionality of this file, you must:
+#   $ pipenv install twine --dev
+
+import io
+import os
+import sys
+from shutil import rmtree
+
+from setuptools import find_packages, setup, Command
+
+# Package meta-data.
+NAME = 'ifkit'
+DESCRIPTION = '日常工作使用的小工具/小功能'
+URL = 'https://www.infuq.com'
+EMAIL = '864511259@qq.com'
+AUTHOR = 'infuq'
+REQUIRES_PYTHON = '>=3.6.0'
+VERSION = '0.1.51'
+
+# What packages are required for this module to be executed?
+REQUIRED = [
+    # 'requests', 'maya', 'records',
+]
+
+# What packages are optional?
+EXTRAS = {
+    # 'fancy feature': ['django'],
+}
+
+# The rest you shouldn't have to touch too much :)
+# ------------------------------------------------
+# Except, perhaps the License and Trove Classifiers!
+# If you do change the License, remember to change the Trove Classifier for that!
+
+here = os.path.abspath(os.path.dirname(__file__))
+
+# Import the README and use it as the long-description.
+# Note: this will only work if 'README.md' is present in your MANIFEST.in file!
+try:
+    with io.open(os.path.join(here, 'README.md'), encoding='utf-8') as f:
+        long_description = '\n' + f.read()
+except FileNotFoundError:
+    long_description = DESCRIPTION
+
+# Load the package's __version__.py module as a dictionary.
+about = {}
+if not VERSION:
+    project_slug = NAME.lower().replace("-", "_").replace(" ", "_")
+    with open(os.path.join(here, project_slug, '__version__.py')) as f:
+        exec(f.read(), about)
+else:
+    about['__version__'] = VERSION
+
+
+class UploadCommand(Command):
+    """Support setup.py upload."""
+
+    description = 'Build and publish the package.'
+    user_options = []
+
+    @staticmethod
+    def status(s):
+        """Prints things in bold."""
+        print('\033[1m{0}\033[0m'.format(s))
+
+    def initialize_options(self):
+        pass
+
+    def finalize_options(self):
+        pass
+
+    def run(self):
+        try:
+            self.status('Removing previous builds…')
+            rmtree(os.path.join(here, 'dist'))
+        except OSError:
+            pass
+
+        self.status('Building Source and Wheel (universal) distribution…')
+        os.system('{0} setup.py sdist bdist_wheel --universal'.format(sys.executable))
+
+        self.status('Uploading the package to PyPI via Twine…')
+        os.system('twine upload dist/*')
+
+        self.status('Pushing git tags…')
+        os.system('git tag v{0}'.format(about['__version__']))
+        os.system('git push --tags')
+
+        sys.exit()
+
+
+# Where the magic happens:
+setup(
+    name=NAME,
+    version=about['__version__'],
+    description=DESCRIPTION,
+    long_description=long_description,
+    long_description_content_type='text/markdown',
+    author=AUTHOR,
+    author_email=EMAIL,
+    python_requires=REQUIRES_PYTHON,
+    url=URL,
+    packages=find_packages(exclude=["tests", "*.tests", "*.tests.*", "tests.*"]),
+    # If your package is a single module, use this instead of 'packages':
+    # py_modules=['mypackage'],
+
+    # entry_points={
+    #     'console_scripts': ['mycli=mymodule:cli'],
+    # },
+    install_requires=REQUIRED,
+    extras_require=EXTRAS,
+    include_package_data=True,
+    license='MIT',
+    classifiers=[
+        # Trove classifiers
+        # Full list: https://pypi.python.org/pypi?%3Aaction=list_classifiers
+        'License :: OSI Approved :: MIT License',
+        'Programming Language :: Python',
+        'Programming Language :: Python :: 3',
+        'Programming Language :: Python :: 3.6',
+        'Programming Language :: Python :: Implementation :: CPython',
+        'Programming Language :: Python :: Implementation :: PyPy'
+    ],
+    # $ setup.py publish support.
+    cmdclass={
+        'upload': UploadCommand,
+    },
+)