FileStream/Database/Elements.py

from FileStream.Tools import Time_ISTKolNow


def NewTG_Files(details:dict) -> dict:
    return {
        "user_id": details['user_id'] if details['user_id'] else None,
        "user_type": details['user_type'] if details['user_type'] else None,
        "message_id": details['message_id'] if details['message_id'] else None,
        "location":details['location'] if details['location'] else None,
        "IMDB_id": "Unknown",
        "poster": "Unknown",   
        "title": "Unknown",
        "type": "Unknown",
        "description":"Unknown",
        "genre":"Unknown",
        "quality":"Unknown",
        "time": details['time'] if details['time'] else None,
        "privacy_type": details['privacy_type'] if details['privacy_type'] else None,
        "file_ids":{},
        "file": {
            "file_id":details['file']['file_id'] if details['file']['file_id'] else None,
            "caption":"",
            "file_unique_id": details['file']['file_unique_id'] if details['file']['file_unique_id'] else None,
            "file_name":details['file']['file_name'] if details['file']['file_name'] else None,
            "file_size":details['file']['file_size'] if details['file']['file_size'] else None,
            "mime_type":details['file']['mime_type']if details['file']['mime_type'] else None,
            "taged_users": {}
        },
    }


def UserSchema(id: str, details) -> dict:

    return dict(
        telegram_id=id,
        access="USER",
        name= details.first_name + details.last_name if details else "Unknown",
        tele_status={
            "status": "ACTIVE",
            "activity": None,
            "joined": Time_ISTKolNow()
        },
        file={
            "links": 0,
            "private_files": 0,
            "public_files": 0,
        },
        site_id="None",
        site_status={
            "status": None,
            "activity": None,
            "password": None,
            "links": 0,
            "joined": "None"
        },
    )

async def ChangeUserAccessType(self, id):
    user= await self.get_user(id)
    if user:
      await self.users.update_one({"_id": user['_id']}, {"$set": {"access":"ADMIN" }})
    else:
      user = UserSchema(id)
      user['access']="ADMIN"
      await self.users.insert_one(user)