HoloUA/Discord
1
0
Fork 0
Code Issues Pull Requests Releases 9 Activity

Formatted with black

Browse Source
This commit is contained in:
Profitroll
2023-05-04 16:14:23 +02:00
parent 83fe753a15
commit cea97f9bad
5 changed files with 149 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

59
classes/holo_user.py
View File

@@ -11,20 +11,28 @@ try:
except ImportError:
from typing_extensions import Literal
class NotEnoughMoneyError(Exception):
"""User does not have enough money to do that"""
pass
class UserNotFoundError(Exception):
"""HoloUser could not find user with such an ID in database"""
def __init__(self, user, user_id):
self.user = user
self.user_id = user_id
super().__init__(f"User of type {type(self.user)} with id {self.user_id} was not found")
super().__init__(
f"User of type {type(self.user)} with id {self.user_id} was not found"
)
class HoloUser():
def __init__(self, user: Union[discord.User, discord.Member, discord.member.Member, int]) -> None:
class HoloUser:
def __init__(
self, user: Union[discord.User, discord.Member, discord.member.Member, int]
) -> None:
"""Get an object that has a proper binding between Discord ID and database
### Args:
@@ -32,10 +40,10 @@ class HoloUser():
### Raises:
* `UserNotFoundError`: User with such ID does not seem to exist in database
"""
"""
if hasattr(user, "id"):
self.id = user.id # type: ignore
self.id = user.id # type: ignore
else:
self.id = user
@@ -64,12 +72,12 @@ class HoloUser():
# ### Args:
# * `amount` (int, optional): Amount of XP points to give. Defaults to 1.
# """
# """
# self.xp += amount
# col_users.update_one(filter={"_id": self.db_id}, update={ "$set": { "xp": self.xp } })
# def xp_level_up(self) -> None:
# """Add 1 to the current XP level"""
# """Add 1 to the current XP level"""
# xp_diff = int(self.xp - self.xp_next)
# xp_next = int(self.xp_next*configGet("multiplier", "leveling")+configGet("addition", "leveling"))
# self.xp = xp_diff
@@ -84,7 +92,7 @@ class HoloUser():
# ### Args:
# * `amount` (int): Amount of currency to be set
# """
# """
# self.balance = amount
# col_users.update_one(filter={"_id": self.db_id}, update={ "$set": { "balance": self.balance } })
@@ -93,7 +101,7 @@ class HoloUser():
# ### Args:
# * `amount` (int): Amount to be added
# """
# """
# self.balance_set(self.balance+amount)
# def balance_take(self, amount: int) -> bool:
@@ -104,7 +112,7 @@ class HoloUser():
# ### Returns:
# * `bool`: True if successful and False if not
# """
# """
# if self.balance >= amount:
# self.balance_set(self.balance-amount)
# return True
@@ -121,7 +129,7 @@ class HoloUser():
# ### Raises:
# * `NotEnoughMoneyError`: Not enough money to perform this transaction
# """
# """
# if self.balance >= amount:
# if isinstance(destination, int):
# destination = HoloUser(destination)
@@ -135,7 +143,7 @@ class HoloUser():
# ### Returns:
# * `int`: Amount of money to be earned
# """
# """
# if self.work_xp >= 100:
# return randint(configGet("min", "work", "level", "4"), configGet("max", "work", "level", "4"))
# elif self.work_xp >= 50:
@@ -153,7 +161,7 @@ class HoloUser():
### Returns:
* `int`: Number of warnings
"""
"""
warns = col_warnings.find_one({"user": self.id})
if warns == None:
return 0
@@ -165,12 +173,15 @@ class HoloUser():
### Args:
* `count` (int, optional): Count of warnings to be added. Defaults to 1.
"""
"""
warns = col_warnings.find_one({"user": self.id})
if warns != None:
col_warnings.update_one(filter={"_id": self.db_id}, update={ "$set": { "warns": warns["warns"]+count } })
col_warnings.update_one(
filter={"_id": self.db_id},
update={"$set": {"warns": warns["warns"] + count}},
)
else:
col_warnings.insert_one(document={ "user": self.id, "warns": count })
col_warnings.insert_one(document={"user": self.id, "warns": count})
logWrite(f"User {self.id} was warned {count} times due to: {reason}")
# def cooldown_go(self, kind: Literal["work", "daily", "weekly", "monthly", "steal"]) -> None:
@@ -178,7 +189,7 @@ class HoloUser():
# ### Args:
# * `kind` (Literal["work", "daily", "weekly", "monthly", "steal"]): Kind of a cooldown
# """
# """
# self.cooldown[kind] = datetime.now(tz=timezone.utc)
# col_users.update_one(filter={"_id": self.db_id}, update={ "$set": { "cooldown": self.cooldown } })
@@ -188,20 +199,22 @@ class HoloUser():
### Args:
* `key` (str): Attribute to be changed
* `value` (Any): Value to set
"""
"""
if not hasattr(self, key):
raise AttributeError()
setattr(self, key, value)
col_users.update_one(filter={"_id": self.db_id}, update={ "$set": { key: value } }, upsert=True)
col_users.update_one(
filter={"_id": self.db_id}, update={"$set": {key: value}}, upsert=True
)
logWrite(f"Set attribute {key} of user {self.id} to {value}")
def purge(self) -> None:
"""Completely remove data from database. Will not remove transactions logs and warnings."""
"""Completely remove data from database. Will not remove transactions logs and warnings."""
col_users.delete_one(filter={"_id": self.db_id})
self.unauthorize()
def unauthorize(self) -> None:
"""Cancel Oauth2 authorization"""
"""Cancel Oauth2 authorization"""
col_authorized.find_one_and_delete({"user": self.id})
# def is_authorized(self) -> bool:
@@ -209,7 +222,7 @@ class HoloUser():
# ### Returns:
# * `bool`: True if yes and False if no
# """
# """
# if configGet("mode") == "secure":
# authorized = col_authorized.find_one({"user": self.id})
# if authorized is not None:
@@ -217,4 +230,4 @@ class HoloUser():
# else:
# return False
# else:
# return True
# return True