youtube-dl-gui/youtube_dl_gui/LogManager.py

#! /usr/bin/env python

import wx

from time import strftime
from .Utils import (
  fix_path,
  file_exist,
  get_filesize
)

LOG_FILENAME = 'log'
LOG_FILESIZE = 524288 # 524288B = 512kB

class LogManager():

    def __init__(self, path, add_time=False):
        self.path = fix_path(path) + LOG_FILENAME
        self.add_time = add_time
        self.init_log()
        self.auto_clear_log()

    def auto_clear_log(self):
        if self.size() > LOG_FILESIZE:
            self.clear()

    def init_log(self):
        if not file_exist(self.path):
            self.clear()

    def size(self):
        return get_filesize(self.path)

    def clear(self):
        with open(self.path, 'w') as fl:
            fl.write('')

    def write(self, data):
        with open(self.path, 'a') as fl:
            if self.add_time:
                t = '[%s] ' % strftime('%c')
                fl.write(t)
            fl.write(data)
            fl.write('\n')

class LogGUI(wx.Frame):

    title = 'Log Viewer'

    def __init__(self, path, parent=None, id=-1):
        wx.Frame.__init__(self, parent, id, self.title, size=(650, 200))

        panel = wx.Panel(self)
        textArea = wx.TextCtrl(panel, -1, style = wx.TE_MULTILINE | wx.TE_READONLY | wx.HSCROLL)
        sizer = wx.BoxSizer()
        sizer.Add(textArea, 1, wx.EXPAND)
        panel.SetSizerAndFit(sizer)

        textArea.LoadFile(path)