107 lines
3.4 KiB
Python
107 lines
3.4 KiB
Python
|
# -*- 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
|