forked from RobotTrick/Archive-bot
-
Notifications
You must be signed in to change notification settings - Fork 37
/
utils.py
58 lines (41 loc) · 1.78 KB
/
utils.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
from pony.orm import *
from pyrogram.types import Message
from os import listdir
# ========= DB build =========
db = Database()
class User(db.Entity):
uid = PrimaryKey(int, auto=True)
status = Required(int) # status-user: "INSERT"/"NOT-INSERT"
db.bind(provider='sqlite', filename='zipbot.sqlite', create_db=True)
db.generate_mapping(create_tables=True)
# ========= helping func =========
def dir_work(uid: int) -> str:
""" static-user folder """
return f"static/{uid}/"
def zip_work(uid: int) -> str:
""" zip-archive file """
return f'static/{uid}.zip'
def list_dir(uid: int) -> list:
""" items in static-user folder """
return listdir(dir_work(uid))
def up_progress(current, total, msg: Message):
""" edit status-msg with progress of the uploading """
msg.edit(f"**Upload progress: {current * 100 / total:.1f}%**")
# ========= MSG class =========
class Msg:
def start(msg: Message) -> str:
""" return start-message text """
txt = f"Hey {msg.from_user.mention}!\n" \
"\nI can compress files in to an archive." \
"\nJust send /zip, and follow the instructions."
return txt
zip = "Send the files you want to compress, and at the end send /stopzip after all the files have been downloaded.\n" \
"\n\nNote: due to upload limit, the total size of the file(s) can be at most 2GB."
too_big = "Note: due to upload limit, the total size of the file(s) can be at most 2GB."
too_much = "Note: the total number of the files can be at most 500"
send_zip = "Send /zip to compress the files"
zipping = "start compressing {} files..."
uploading = "uploading archive..."
unknow_error = "An unknown error occurred"
downloading = "downloading..."
zero_files = "No files were sent"