pip install --upgrade prestool
from prestool.Tool import Tool
tool = Tool()
tool.random_name() # 随机姓名
tool.random_phone() # 随机手机号
tool.random_ssn() # 随机身份证
tool.random_string(16) # 随机位数的字符串
tool.random_number(8) # 随机位数的数字
tool.random_ua() # 随机UA
tool.random_ua('chrome') # 随机UA-Chrome
tool.random_ua('firefox') # 随机UA-Firefox
tool.random_ua('ie') # 随机UA-IE
tool.random_ua('opera') # 随机UA-opera
tool.random_ua('safari') # 随机UA-safari
tool.url_encode('编码前的url地址') # 编码
tool.url_decode('解码前的url地址') # 解码
tool.base_64_encode('编码前的字符串') # base64编码
tool.to_md5('原始字符串')
tool.to_hmac_256('原始字符串', '加密key')
tool.to_sha_256('原始字符串')
tool.ding_talk_token = '钉钉机器人token'
tool.ding_talk_sign_key = '钉钉机器人签名key'
tool.send_ding_talk_msg('消息内容')
tool.qy_wechat_token = '企业微信机器人token'
tool.send_qy_wechat_msg('消息内容')
tool.feishu_token = '飞书机器人token'
tool.feishu_sign_key = '飞书机器人秘钥'
tool.send_feishu_msg('消息内容')
tool.mail_from_user_host = '发件地址host'
tool.mail_from_user = '发件人邮箱号'
tool.mail_from_user_pwd = '发件人密码'
tool.send_mail_msg(to_user='收件人邮箱地址(列表)', title='邮件标题', content='邮件内容')
tool.time_stamp() # 秒级时间戳10位
tool.time_stamp('ms') # 毫秒级时间戳13位
tool.get_now_time() # 获取当前时间 20201206000000
tool.get_now_time('-') # 获取当前时间 2020-12-06 00:00:00
tool.date_to_time_stamp('2012-01-01 00:00:00') # 时间字符串转为时间戳
tool.time_stamp_to_date(1732312234) # 时间戳转为时间字符串
tool.json_dumps({"test": "python字典"}) # 字典转json
tool.json_loads('{"test": "python字典"}') # json转字典
tool.xml_to_dict('<xml><data>字符串</data></xml>') # xml转成python字典
tool.dict_to_xml({"test": "python字典"}) # python字典 转成xml
http请求tool.http_client(url='', data={}, method='GET') # get请求
tool.http_client(url='', data={}, method='POST') # post请求
tool.get_cookies(url='接口地址', data={}, method='GET')
tool.get_cookies(url='接口地址', data={}, method='POST')
tool.trans_data_to_url(url='接口地址', data={}) # 把参数拼接到url上
dubbo接口tool.dubbo_args('参数1', '参数2', '参数3') # dubbo接口参数
tool.invoke_dubbo('地址', '端口', '服务API名', '接口方法名', 'dubbo接口参数') # 请求dubbo接口
tool.logger('日志信息')
tool.get_ip_by_url('url地址') # 获取ip
MySQL/Sqlite)sql语句from prestool.PresMySql import SqlStr
sql = SqlStr()
target不传时,为全部字段,即*,where={'key':'value'}
sql.select_sql_str(table='table1', where={'id': 1, 'name': '张三'})
select * from table1 where id = 1 and name = '张三';
target=[i1,i2,i3]时,为相应字段sql.select_sql_str(table='table1', target=['a', 'b', 'c'], where={'id': 1, 'name': '张三'})
select a, b, c from table1 where 1 = 1 and id = 1 and name = '张三';
limit=10 limit='10,1000'为筛选限制字段sql.select_sql_str(
table='table1',
target=['a', 'b', 'c'],
order={'age': 'desc', 'score': 'desc'},
limit=20)
select a, b, c from table1 where 1 = 1 order by age desc, score desc limit 20;
where条件中有的字段为null或者not null时sql.select_sql_str(
table='table1',
target=['a', 'b', 'c'],
where={'id': 1},
is_not_null={'age': True, 'name': None})
select a, b, c from table1 where 1 = 1 and id = 1 and age is not null and name is null;
sql.select_sql_str(
table='table1',
target=['a', 'b', 'c'],
order={'age': 'desc', 'score': 'desc'})
select a, b, c from table1 order by age desc, score desc;
in语句sql.select_sql_str(
table='table1',
target=['a', 'b', 'c'],
select_in={'orders': [123121312, 123123445, 213123]})
not in语句sql.select_sql_str(
table='table1',
target=['a', 'b', 'c'],
select_not_in={'orders': [123121312, 123123445, 213123]})
select a, b, c from table1 where 1 = 1 and orders not in (123121312, 123123445, 213123);
like语句sql.select_sql_str(table='table1', target=['a', 'b', 'c'], like={'name': '%光', 'address': "中国%"})
select a, b, c from table1 where 1 = 1 and name like '%光' and address like '中国%';
between语句sql.select_sql_str(
table='table1',
target=['a', 'b', 'c'],
between={'age': (10, 20), 'year': (2021, 2022)})
select a, b, c from table1 where 1 = 1 and age between 10 and 20 and year between 2021 and 2022;
sql.select_sql_str(
table='table1',
target=['a', 'b', 'c'],
compare={'age': {'>': 10, '<': 20}, 'year': {'>=': '2021'}})
select a, b, c from table1 where 1 = 1 and age > 10 and age < 20 and year >= 2021;
target为要更新的数据,为字典结构 (支持大于、小于语句、between 语句、like 语句、in 语句)sql.update_sql_str(table='table1', target={'name': '李四', 'age': 15}, where={'id': 1, 'name': '张三'})
update table1 set name='李四',age=15 where id = 1 and name = '张三';
sql.delete_sql_str(table='table1', where={'id': 1, 'name': '张三'})
delete from table1 where id = 1 and name = '张三';
sql.insert_sql_str(table='table1', target={'id': 1, 'name': '张三'})
insert into table1 (id, name) values (1, '张三');
mysql模式from prestool.PresMySql import PresMySql
pres = PresMySql()
sqlite模式from prestool.PresSqlite import PresSqlite
pres = PresSqlite()
pres.mysql_host = ''
pres.mysql_port = 3306
pres.mysql_user = ''
pres.mysql_pwd = ''
pres.mysql_db_name = ''
pres.mysql_charset = 'utf8mb4'
pres.sqlite_path = ''
pres.to_query(table='table1', target=['a', 'b', 'c'], between={'age': (10, 20), 'year': (2021, 2022)})
pres.to_insert(table='table1', target={'id': 1, 'name': '张三'})
pres.to_delete(table='table1', where={'id': 1, 'name': '张三'})
pres.to_update(table='table1', target={'name': '李四', 'age': 15}, where={'id': 1, 'name': '张三'})
from prestool.PresMySql import SqlStr
sql = SqlStr()
# INSERT 参数化
sql_str, values = sql.insert_sql_str_params(table='table1', target={'id': 1, 'name': "O'Brien"})
# 返回: ('INSERT INTO table1 (id, name) VALUES (%s, %s)', [1, "O'Brien"])
# UPDATE 参数化
sql_str, values = sql.update_sql_str_params(table='table1', target={'name': "O'Brien"}, where={'id': 1})
# 返回: ('UPDATE table1 SET name = %s WHERE id = %s', ["O'Brien", 1])
# DELETE 参数化
sql_str, values = sql.delete_sql_str_params(table='table1', where={'id': 1, 'name': "O'Brien"})
# 返回: ('DELETE FROM table1 WHERE id = %s AND name = %s', [1, "O'Brien"])
from prestool.PresMySql import PresMySql
pres = PresMySql()
# ... 初始化数据库连接 ...
# 参数化 INSERT(自动处理特殊字符,防止 SQL 注入)
pres.to_insert_params(table='table1', target={'id': 1, 'name': "O'Brien"})
# 参数化 UPDATE
pres.to_update_params(table='table1', target={'name': "O'Brien"}, where={'id': 1})
# 参数化 DELETE
pres.to_delete_params(table='table1', where={'id': 1, 'name': "O'Brien"})
# 自定义参数化 SQL
pres.exec_sql('SELECT * FROM table1 WHERE name = %s', select='all', args=["O'Brien"])
注意:
to_query 方法始终返回 list[dict],空结果返回 []
%s 作为占位符(SQLite 会自动转换为 ?)MongoDB)from prestool.PresMongo import PresMongo
pres = PresMongo()
pres.mongo_host = ''
pres.mongo_port = 27017
pres.mongo_user = ''
pres.mongo_pwd = ''
pres.mongo_db_name = ''
pres.mongo_auth_source = ''
pres.to_query(table='abc', where={'id': "123123"})
pres.to_query(table='abc', where={'$and': [{'age': 14}, {'sex': 1}]})
res = pres.to_query(table='abc', where={'id': "123123"}, is_all=True)
for i in res:
print(i)
pres.to_query(table='abc', where={'id': "123123"}, asc='age') # 按年龄正序
pres.to_query(table='abc', where={'id': "123123"}, desc='age') # 按年龄倒序
pres.to_query(table='abc', where={'id': "123123"}, limit=10)
pres.to_update(table='abc', target={'age': 14}, where={'name': '张三'})
pres.to_insert(table='abc', target={'age': 14}) # 插入一条
pres.to_insert(table='abc', target=[{'age': 14}, {"name": '张三'}]) # 插入一个列表
Redis)Redis
from prestool.PresRedis import PresRedis
pres = PresRedis()
redis
pres.redis_host = 'localhost' # Redis 地址
pres.redis_port = 6379 # Redis 端口
pres.redis_password = None # 密码(可选)
pres.redis_db = 0 # 数据库编号(可选)
# 设置值(ex=过期秒数)
pres.set('key', 'value', ex=3600)
# 获取值
value = pres.get('key')
# 检查键是否存在
exists = pres.exists('key')
# 删除键
pres.delete('key')
# 设置过期时间
pres.expire('key', 7200)
# 获取剩余 TTL
ttl = pres.ttl('key')
# 原子递增/递减
pres.incr('counter') # +1
pres.decr('counter', 5) # -5
# 存储 JSON(自动序列化)
data = {'name': '张三', 'age': 25, 'roles': ['admin', 'editor']}
pres.set_json('user:1', data, ex=7200)
# 读取 JSON(自动反序列化)
result = pres.get_json('user:1')
# 设置字段
pres.hset('user:profile', 'nickname', '张三')
pres.hset('user:profile', 'level', 'VIP')
# 获取单个字段
nickname = pres.hget('user:profile', 'nickname')
# 批量设置
pres.hmset('user:profile', {'phone': '13800138000', 'email': 'test@example.com'})
# 获取所有字段
profile = pres.hgetall('user:profile')
# 返回: {'nickname': '张三', 'level': 'VIP', 'phone': '13800138000', 'email': 'test@example.com'}
# 检查字段存在
has_level = pres.hexists('user:profile', 'level')
# 获取字段数量
field_count = pres.hlen('user:profile')
# 获取所有字段名/字段值
fields = pres.hkeys('user:profile')
values = pres.hvals('user:profile')
# 删除字段
pres.hdel('user:profile', 'level')
# 头部/尾部插入
pres.lpush('queue', 'task1', 'task2')
pres.rpush('queue', 'task3')
# 头部/尾部弹出
task = pres.lpop('queue')
task = pres.rpop('queue')
# 获取列表长度
length = pres.llen('queue')
# 获取范围元素
items = pres.lrange('queue', 0, 9) # 前10个
# 添加成员
pres.sadd('tags', 'python', 'redis', 'cache')
# 获取所有成员
members = pres.smembers('tags')
# 检查成员
is_member = pres.sismember('tags', 'python')
# 移除成员
pres.srem('tags', 'cache')
# 成员数量
count = pres.scard('tags')
# 按模式扫描删除(推荐)
pres.scan_delete('cache:*')
pres.scan_delete('session:*')
# 获取匹配的所有键(慎用,影响性能)
keys = pres.keys('user:*')
# 迭代获取键(生产环境推荐)
for key in pres.scan_iter('cache:*'):
print(key)
# 清空当前数据库(危险!)
pres.flushdb()
# 直接使用 StrictRedis 实例
with pres.conn_redis() as r:
r.set('raw:key', 'raw:value')
r.get('raw:key')
uv build
uv publish