Initial commit

modules/functions.py

@@ -0,0 +1,107 @@
+# -*- coding: utf-8 -*-
+"""Some set of functions needed for discord/telegram bots and other types of apps"""
+
+import gzip
+import os
+import shutil
+
+from datetime import datetime
+from ujson import loads, dumps
+
+
+def nowtimeGet(format="%H:%M:%S | %d.%m.%Y"):
+    """Return current local time formatted as arg.
+
+    ### Args:
+        * format (str, optional): Format that should be returned. Defaults to "%H:%M:%S | %d.%m.%Y".
+
+    ### Returns:
+        * str: Local time formatted as arg.
+    """    
+    return datetime.now().strftime(format)
+
+
+def checkSize(logs_folder="logs/", log_size=1024):
+    """Checks latest log file size and rotates it if needed.
+
+    ### Args:
+        * logs_folder (str, optional): Folder where logs stored. Defaults to "logs/".
+        * log_size (int, optional): How many bytes should file containt to be rotated. Defaults to 1024.
+    """    
+    i = 0
+    while i < 2:
+        try:
+            log = os.stat(logs_folder + 'latest.log')
+            if (log.st_size / 1024) > log_size:
+                with open(logs_folder + 'latest.log', 'rb') as f_in:
+                    with gzip.open(f'{logs_folder}{datetime.now().strftime("%d.%m.%Y_%H:%M:%S")}.zip', 'wb') as f_out:
+                        shutil.copyfileobj(f_in, f_out)  
+                open(logs_folder + 'latest.log', 'w').close()
+            i = 2
+        except FileNotFoundError:
+            try:
+                log = open(logs_folder + 'latest.log', 'a')
+                open(logs_folder + 'latest.log', 'a').close()
+            except:
+                try:
+                    os.mkdir(logs_folder)
+                    log = open(logs_folder + 'latest.log', 'a')
+                    open(logs_folder + 'latest.log', 'a').close()
+                except:
+                    pass
+            i += 1
+
+
+def logWrite(message, logs_folder="logs/", level="INFO"):
+    """Append some message to latest log file.
+
+    ### Args:
+        * message (str): Something you want to add to log
+        * logs_folder (str, optional): Folder where logs stored. Defaults to "logs/".
+        * level (str, optional): Log level (INFO, WARN, ERRO, CRIT, DEBG). Defaults to "INFO".
+    """    
+
+    checkSize(logs_folder=logs_folder)
+
+    try:
+        log = open(logs_folder + 'latest.log', 'a')
+        open(logs_folder + 'latest.log', 'a').close()
+    except:
+        try:
+            os.mkdir(logs_folder)
+            log = open(logs_folder + 'latest.log', 'a')
+            open(logs_folder + 'latest.log', 'a').close()
+        except:
+            print(f'[{nowtimeGet()}] [ERRO] Log file could not be written.', flush=True)
+            return
+
+    log.write(f'[{nowtimeGet()}] [{level}] {message}\n')
+    print(f'[{nowtimeGet()}] [{level}] {message}', flush=True)
+    log.close()
+
+
+def jsonSave(filename, value):
+    """Save some list or dict as json file.
+
+    Args:
+        * filename (str): File to which value will be written.
+        * value (list or dict): Some object that will be written to filename.
+    """    
+    with open(filename, 'w', encoding="utf-8") as f:
+        f.write(dumps(value, indent=4, ensure_ascii=False))
+        f.close()
+
+
+def jsonLoad(filename):
+    """Load json file and return python dict or list.
+
+    Args:
+        * filename (str): File which should be loaded.
+
+    Returns:
+        * list or dict: Content of json file provided.
+    """    
+    with open(filename, 'r', encoding="utf-8") as f:
+        value = loads(f.read())
+        f.close()
+    return value

modules/functions_bot.py

@@ -0,0 +1,45 @@
+# -*- coding: utf-8 -*-
+from modules.functions import jsonLoad, jsonSave
+
+def locale(key, *args):
+    strings = jsonLoad("strings.json")
+    string = strings
+    for dict_key in args:
+        string = string[dict_key]
+    return string[key]
+
+
+def configGet(key):
+    return jsonLoad("config.json")[key]
+
+def configAppend(key, value):
+    config = jsonLoad("config.json")
+    config[key].append(value)
+    jsonSave("config.json", config)
+
+def configRemove(key, value):
+    config = jsonLoad("config.json")
+    config[key].remove(value)
+    jsonSave("config.json", config)
+
+
+def userSet(userid, key, value):
+    user = jsonLoad(f"data/users/{userid}.json")
+    user[key] = value
+    jsonSave(f"data/users/{userid}.json", user)
+
+def userGet(userid, key):
+    try:
+        return jsonLoad(f"data/users/{userid}.json")[key]
+    except KeyError:
+        return None
+    except FileNotFoundError:
+        return None
+
+def userClear(userid, key):
+    try:
+        user = jsonLoad(f"data/users/{userid}.json")
+        del user[key]
+        jsonSave(f"data/users/{userid}.json", user)
+    except KeyError:
+        pass