first version

for_db.py

@@ -0,0 +1,171 @@
+import yaml
+from datetime import datetime
+from telebot import types
+import errno, os
+
+users_path = "database/users.yaml"
+user_path = "database/users/{}.yaml"
+
+
+def check_user(user_id: int) -> dict:
+    users = yaml.safe_load(open(users_path, "r", encoding="utf-8"))
+
+    if not users:
+        users = {}
+
+    if user_id in users:
+        return users[user_id]
+
+    return None
+
+
+def new_user(message: types.Message) -> bool:
+    if check_user(message.chat.id):
+        return False
+
+    user = {
+        "id": message.chat.id,
+        "type": message.chat.type,
+        "username": message.chat.username,
+        "first_name": message.chat.first_name,
+        "last_name": message.chat.last_name,
+        "time_created": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+    }
+
+    users = yaml.safe_load(open(users_path, "r"))
+
+    if not users:
+        users = {}
+
+    users[user["id"]] = user
+
+    yaml_data = yaml.dump(
+        users,
+        default_flow_style=False,
+        encoding="utf-8",
+        allow_unicode=True,
+        width=float("inf"),
+        sort_keys=False,
+    )
+
+    with open(users_path, "wb") as file:
+        file.write(yaml_data)
+
+    return True
+
+
+def add_note(id: int, text: str, time_notif: datetime = None, list: str = "Default"):
+    if not check_user(id):
+        return False
+
+    notes = (
+        yaml.safe_load(open(user_path.format(id), "r", encoding="utf-8"))
+        if os.path.exists(user_path.format(id))
+        else {}
+    )
+
+    if list not in notes:
+        notes[list] = {"name": list, "description": "Стандартный список", "notes": []}
+
+    notes[list]["notes"].append(
+        {
+            "text": text,
+            "time_notif": str(time_notif) if time_notif else None,
+            "time_created": str(datetime.now().strftime("%Y-%m-%d %H:%M:%S")),
+        }
+    )
+
+    if True:  # TODO: edit sorting
+        import functools
+
+        def comp(x, y):
+            if not x["time_notif"] and not y["time_notif"]:
+                return 0
+            elif not x["time_notif"]:
+                return -1
+            elif not y["time_notif"]:
+                return 1
+            else:
+                x_time = datetime.strptime(
+                    x["time_notif"],
+                    "%Y-%m-%d %H:%M:%S",
+                )
+                y_time = datetime.strptime(
+                    y["time_notif"],
+                    "%Y-%m-%d %H:%M:%S",
+                )
+                if x_time > y_time:
+                    return 1
+                elif x_time < y_time:
+                    return -1
+            return 0
+
+        notes[list]["notes"].sort(
+            key=lambda x: datetime.strptime(
+                x["time_created"],
+                "%Y-%m-%d %H:%M:%S",
+            ),
+            reverse=True,
+        )
+
+        notes[list]["notes"].sort(
+            key=functools.cmp_to_key(comp),
+        )
+
+    yaml_data = yaml.dump(
+        notes,
+        default_flow_style=False,
+        encoding="utf-8",
+        allow_unicode=True,
+        width=float("inf"),
+        sort_keys=False,
+    )
+
+    with open(user_path.format(id), "wb") as file:
+        file.write(yaml_data)
+
+    return True
+
+
+def get_notes(id, list="Default"):
+    if not check_user(id):
+        return None
+
+    if not os.path.exists(user_path.format(id)):
+        return None
+
+    notes = yaml.safe_load(open(user_path.format(id), "r", encoding="utf-8"))
+
+    return notes[list] if list in notes else None
+
+
+def delete_note(id, note_ind, list="Default"):
+    if not check_user(id):
+        return False
+
+    if not os.path.exists(user_path.format(id)):
+        return False
+
+    notes = yaml.safe_load(open(user_path.format(id), "r", encoding="utf-8"))
+
+    if list not in notes:
+        return False
+
+    if note_ind >= len(notes[list]["notes"]):
+        return False
+
+    notes[list]["notes"].pop(note_ind)
+
+    yaml_data = yaml.dump(
+        notes,
+        default_flow_style=False,
+        encoding="utf-8",
+        allow_unicode=True,
+        width=float("inf"),
+        sort_keys=False,
+    )
+
+    with open(user_path.format(id), "wb") as file:
+        file.write(yaml_data)
+
+    return True