mirror of
https://github.com/tgbot-collection/YYeTsBot.git
synced 2025-11-25 11:29:38 +08:00
72 lines
1.6 KiB
Python
72 lines
1.6 KiB
Python
# coding: utf-8
|
|
# YYeTsBot - utils.py
|
|
# 2019/8/15 20:27
|
|
|
|
__author__ = 'Benny <benny.think@gmail.com>'
|
|
|
|
import os
|
|
import sys
|
|
import pickle
|
|
import json
|
|
import logging
|
|
import requests
|
|
import redis
|
|
|
|
from config import AJAX_LOGIN, USERNAME, PASSWORD, REDIS
|
|
|
|
r = redis.StrictRedis(host=REDIS, decode_responses=True)
|
|
|
|
cookie_file = os.path.join(os.path.dirname(__file__), 'data', 'cookies.dump')
|
|
|
|
|
|
|
|
|
|
def save_error_dump(uid, err: str):
|
|
r.set(uid, err)
|
|
|
|
|
|
def get_error_dump(uid) -> str:
|
|
err = r.get(uid)
|
|
r.delete(uid)
|
|
if not err:
|
|
err = ""
|
|
return err
|
|
|
|
|
|
def redis_announcement(content="", op="get"):
|
|
if op == "get":
|
|
return r.get("announcement")
|
|
elif op == "set":
|
|
r.set("announcement", content)
|
|
elif op == "del":
|
|
r.delete("announcement")
|
|
|
|
|
|
def today_request(request_type: str):
|
|
if r.exists("usage"):
|
|
data: str = r.get("usage")
|
|
dict_data: dict = json.loads(data)
|
|
dict_data[request_type] += 1
|
|
saved_data: str = json.dumps(dict_data)
|
|
else:
|
|
data_format: dict = dict(total=0, invalid=0, answer=0, success=0, fail=0)
|
|
data_format[request_type] += 1
|
|
saved_data: str = json.dumps(data_format)
|
|
|
|
r.set("usage", saved_data)
|
|
|
|
|
|
def reset_request():
|
|
r.delete("usage")
|
|
|
|
|
|
def show_usage():
|
|
m = "今天我已经服务了{total}次🤓，无效请求{invalid}😆，主人回复{answer}次🤨，成功请求{success}次😝，失败请求{fail}次🤣"
|
|
data: str = r.get("usage")
|
|
if r.exists("usage"):
|
|
dict_data: dict = json.loads(data)
|
|
else:
|
|
dict_data: dict = dict(total=0, invalid=0, answer=0, success=0, fail=0)
|
|
|
|
return m.format(**dict_data)
|