main_form.py

#! /usr/bin/env python
# -*- coding: utf-8 -*-

import tkinter
from tkinter.filedialog import askopenfilename
from tkinter import messagebox
import time
import threading
import os
import gc
import mimetypes
import logging
from UI_modules.custom_notebook import CustomNotebook
from UI_modules.custom_tab import Tab
from UI_modules.window_settings import WindowSetting
from modules.saver import Saver
from modules.list_of_tab import list_of_tab

logging.basicConfig(format='%(asctime)s:[%(levelname)s] %(message)s',
                    filename='logviewer_debug.log',
                    level=logging.DEBUG)

APP_WIDTH_WIN = 750
APP_WIDTH_LINUX = 825
APP_HEIGHT = 400


# список всех вкладок
list_of_tab = list_of_tab


# диалоговое окно открытия файла, возвращает путь к файлу
def path_to_file():
    op = askopenfilename(defaultextension='.log',
                         filetypes=(("Log files", ".log"),
                                    ("Text files", ".txt"),
                                    ("All files", ".*")))
    logging.info('Open file {0}'.format(op))
    return op


# функция сохранения активной вкладки
def save_file(event=None):
    tabs = list_of_tab.get_all_tab()
    tab_name = nb.tab(nb.select(), 'text')
    saver = Saver(tabs, tab_name)
    saver.save_one()
    logging.info('Save text from tab {0}'.format(tab_name))


# функция сохранения всех вкладок
def save_all_file(event=None):
    tabs = list_of_tab.get_all_tab()
    saver = Saver(tabs)
    saver.save_all()
    logging.info('Save all text from all {0} tabs '.format(len(tabs)))


# обновление информации на всех вкладках
def update_tabs():
    while True:
        for tab in list_of_tab.get_all_tab():
            try:
                tab.update_text()
            except AttributeError:
                logging.warning("Update_tabs: Tab is not existing")
        time.sleep(1)


# добавление вкладки
def add_tab(event=None):
    file_path = path_to_file()
    if file_path:
        type_of_file = mimetypes.guess_type(file_path)

        if type_of_file[0]:
            type_of_file = type_of_file[0].split('/')[0]

        if type_of_file == 'text' or not type_of_file[0]:
            list_of_tab.add_tab(Tab(nb, file_path))
            logging.debug('Added tab for file %s', file_path)
            del file_path
            del type_of_file

        else:
            file_name = file_path.split('/')[-1]
            logging.warning('%s is not a text', file_name)
            messagebox.showinfo('Information', '{0} is not a text!'.format(file_name))
            del file_path
            del type_of_file
    else:
        del file_path
        return


def window_settings(event=None):
    window = WindowSetting()
    window.show()


logging.debug('-------------------Programm start-------------------')
if os.name == 'posix':
    logging.debug('This is Linux OS')
    app_width = APP_WIDTH_LINUX
    path_to_icon = ''
else:
    logging.debug('This is Windows OS')
    app_width = APP_WIDTH_WIN
    icon = 'icons\icon.ico'
    path_to_icon = os.path.join(os.getcwd(), icon)

# здесь начинается описание UI
gc.enable()
root = tkinter.Tk()

app_height = APP_HEIGHT
start_pos_x = int(root.winfo_screenwidth() / 2) - int((app_width / 2))
start_pos_y = int(root.winfo_screenheight() / 2.5) - int((app_height / 2))

menu_bar = tkinter.Menu(root, tearoff=False)
save_button = tkinter.Menu(root)

root.config(menu=menu_bar)

submenu = tkinter.Menu(menu_bar, tearoff=False)
save_submenu = tkinter.Menu(submenu, tearoff=False)

menu_bar.add_cascade(label='File', menu=submenu)
submenu.add_command(label="Open file", command=add_tab)
submenu.add_cascade(label='Save..', menu=save_submenu)
save_submenu.add_command(label='Save active', command=save_file)
save_submenu.add_command(label='Save all...', command=save_all_file)

menu_bar.add_command(label='Settings..', command=window_settings)

root.bind('<Control-KeyPress-o>', add_tab)
root.bind('<Control-KeyPress-s>', save_file)
root.bind('<Control-Shift-KeyPress-S>', save_all_file)

# Определение и расположение на форме виджета Блокнот
nb = CustomNotebook(root)
nb.enable_traversal()
nb.pack(fill='both', expand='yes')
# здесь заканчивается описание UI

# поток для обновления вкладок
thread_update_tabs = threading.Thread(target=update_tabs, daemon=True, name='update_tabs')
thread_update_tabs.start()

root.title('LogViewer')
root.iconbitmap(path_to_icon)
root.minsize(app_width, app_height)
root.geometry('{0}x{1}'.format(app_width, app_height))
root.geometry('+{0}+{1}'.format(start_pos_x, start_pos_y))

# запуск отрисовки UI
root.mainloop()
logging.debug('-------------------Programm close-------------------')