"""
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=")])