wttr.in/lib/proxy_log.py
2022-10-25 19:16:48 +02:00

46 lines
939 B
Python

"""
Logger of proxy queries
"""
# pylint: disable=consider-using-with,too-few-public-methods
import datetime
class Logger:
"""
Generic logger.
For specific loggers, _shorten_query() should be rewritten.
"""
def __init__(self, filename):
self._filename = filename
self._file = open(filename, "a", encoding="utf-8")
def _shorten_query(self, query):
return query
def log(self, query, error):
"""
Log `query` and `error`
"""
message = str(datetime.datetime.now())
query = self._shorten_query(query)
if error != "":
message += " ERR " + query + " " + error
else:
message = " OK " + query
self._file.write(message+"\n")
class LoggerWWO(Logger):
"""
WWO logger.
"""
def _shorten_query(self, query):
return "".join([x for x in query.split("&") if x.startswith("q=")])