mirror of
https://github.com/partitioncloud/partitioncloud-server.git
synced 2025-01-23 17:26:26 +01:00
74 lines
2.3 KiB
Python
74 lines
2.3 KiB
Python
from datetime import datetime
|
|
from typing import Union
|
|
from enum import Enum
|
|
|
|
global log_file
|
|
global enabled
|
|
|
|
|
|
class LogEntry(Enum):
|
|
LOGIN = 1
|
|
NEW_GROUPE = 2
|
|
NEW_ALBUM = 3
|
|
NEW_PARTITION = 4
|
|
NEW_USER = 5
|
|
SERVER_RESTART = 6
|
|
FAILED_LOGIN = 7
|
|
|
|
def from_string(entry: str):
|
|
mapping = {
|
|
"LOGIN": LogEntry.LOGIN,
|
|
"NEW_GROUPE": LogEntry.NEW_GROUPE,
|
|
"NEW_ALBUM": LogEntry.NEW_ALBUM,
|
|
"NEW_PARTITION": LogEntry.NEW_PARTITION,
|
|
"NEW_USER": LogEntry.NEW_USER,
|
|
"SERVER_RESTART": LogEntry.SERVER_RESTART,
|
|
"FAILED_LOGIN": LogEntry.FAILED_LOGIN
|
|
}
|
|
# Will return KeyError if not available
|
|
return mapping[entry]
|
|
|
|
|
|
def add_entry(entry: str) -> None:
|
|
date = datetime.now().strftime('%y-%b-%Y %H:%M:%S')
|
|
|
|
with open(log_file, 'a', encoding="utf8") as f:
|
|
f.write(f"[{date}] {entry}\n")
|
|
|
|
|
|
def log(content: list[Union[str, bool, int]], log_type: LogEntry) -> None:
|
|
description: str = ""
|
|
|
|
if log_type not in enabled:
|
|
return
|
|
|
|
match log_type:
|
|
case LogEntry.LOGIN: # content = (user.name)
|
|
description = f"Successful login for {content[0]}"
|
|
|
|
case LogEntry.NEW_GROUPE: # content = (groupe.name, groupe.id, user.name)
|
|
description = f"{content[2]} added groupe '{content[0]}' ({content[1]})"
|
|
|
|
case LogEntry.NEW_ALBUM: # content = (album.name, album.id, user.name)
|
|
description = f"{content[2]} added album '{content[0]}' ({content[1]})"
|
|
|
|
case LogEntry.NEW_PARTITION: # content = (partition.name, partition.uuid, user.name)
|
|
description = f"{content[2]} added partition '{content[0]}' ({content[1]})"
|
|
|
|
case LogEntry.NEW_USER: # content = (user.name, user.id, from_register_page, admin.name if relevant)
|
|
if not content[2]:
|
|
description = f"New user {content[0]}[{content[1]}]"
|
|
else:
|
|
description = f"New user {content[0]}[{content[1]}] added by {content[3]}"
|
|
|
|
case LogEntry.SERVER_RESTART: # content = ()
|
|
description = "Server just restarted"
|
|
|
|
case LogEntry.FAILED_LOGIN: # content = (user.name)
|
|
description = f"Failed login for {content[0]}"
|
|
|
|
add_entry(description)
|
|
|
|
|
|
log_file = "logs.txt"
|
|
enabled = [i for i in LogEntry] |