GUI.py

from tkinter import *
from functools import partial
from tkinter.font import Font

from pynput.mouse import Listener, Button as Btn

from Controller import Controller
from TravianActions import TravianActions
from TravianContract import TroopsRaidType, TroopsType, FarmType, TroopType


def main():
    TravianBotApp().mainloop()


class TravianBotApp(Tk):
    travian_controller = Controller()

    def __init__(self, *args, **kwargs):
        super().__init__()
        Tk.__init__(self, *args, **kwargs)

        self.title_font = Font(family='Helvetica', size=18, weight="bold", slant="italic")

        # the container is where we'll stack a bunch of frames
        # on top of each other, then the one we want visible
        # will be raised above the others
        self.container = Frame(self)
        self.container.grid_rowconfigure(0, weight=1)
        self.container.grid_columnconfigure(0, weight=1)

        # self.frames = {}
        # for F in [LoginPage]:
        #     page_name = F.__name__
        #     frame = F(parent=self.container, controller=self, travian_controller=self.travian_controller)
        #     self.frames[page_name] = frame
        #
        #     # put all of the pages in the same location;
        #     # the one on the top of the stacking order
        #     # will be the one that is visible.
        #     frame.grid(row=0, column=0, sticky="nsew")

        self.show_frame("LoginPage")

    def show_frame(self, page_name):
        """Show a frame for the given page name"""
        frame = None
        if page_name == "LoginPage":
            frame = LoginPage(parent=self.container, controller=self, travian_controller=self.travian_controller)
        elif page_name == "MainPage":
            frame = MainPage(parent=self.container, controller=self, travian_controller=self.travian_controller)

        frame.pack()
        frame.tkraise()


def combine_funcs(*funcs):
    def combined_func(*args, **kwargs):
        for f in funcs:
            f(*args, **kwargs)

    return combined_func


class LoginPage(Frame):
    def __init__(self, parent, controller, travian_controller, **kw):
        Frame.__init__(self, parent)
        super().__init__(**kw)
        self.controller = controller
        self.travian_controller = travian_controller

        # username label and text entry box
        Label(self, text="Username").grid(row=0, column=0)
        username = StringVar()
        Entry(self, textvariable=username).grid(row=0, column=1)

        # password label and password entry box
        Label(self, text="Password").grid(row=1, column=0)
        password = StringVar()
        Entry(self, textvariable=password, show='*').grid(row=1, column=1)

        login_action = partial(self.login, username, password)

        # login button
        Button(self, text="Login",
               command=combine_funcs(login_action, lambda: controller.show_frame("MainPage"))).grid(row=4,
                                                                                                    column=0)

    def login(self, username, password):
        self.travian_controller.login(username.get(), password.get())


class MainPage(Frame):
    listener = None

    def __init__(self, parent, controller, travian_controller, **kw):
        Frame.__init__(self, parent)
        super().__init__(**kw)

        self.controller = controller
        self.travian_controller = travian_controller
        troops = self.travian_controller.tribe.get_troops()
        labelframe = LabelFrame(self, text="Record farms")
        switch_variable = StringVar(value="off")
        Radiobutton(labelframe, text="Off", variable=switch_variable,
                    indicatoron=False, value="off", width=8, command=self.unregister_click_listener).grid(
            row=0, column=1)
        Radiobutton(labelframe, text="On", variable=switch_variable,
                    indicatoron=False, value="on", width=8, command=self.register_click_listener).grid(
            row=0, column=2)

        labelframe.grid(row=2, columnspan=7, sticky='WE',
                        padx=5, pady=5, ipadx=5, ipady=5)

        farm_type_auto = IntVar()
        is_stable = BooleanVar()
        is_barracks = BooleanVar()
        is_batch = BooleanVar()
        auto_mode_label_frame = LabelFrame(self, text="Auto mode")
        Checkbutton(auto_mode_label_frame, text="Stable", variable=is_stable,
                    onvalue=True, offvalue=False, height=5,
                    width=20).grid(row=3, column=2)
        Checkbutton(auto_mode_label_frame, text="Barracks", variable=is_barracks,
                    onvalue=True, offvalue=False, height=5,
                    width=20).grid(row=3, column=3)
        Checkbutton(auto_mode_label_frame, text="Batch", variable=is_batch,
                    onvalue=True, offvalue=False, height=5,
                    width=20).grid(row=3, column=4)

        troops_type_label_frame = LabelFrame(auto_mode_label_frame, text="Troops Training Type")
        infantry_troops_training_type_auto = StringVar()

        infantry_troops = self.travian_controller.tribe.get_troops_by_type(troops_type=TroopType.INFANTRY)
        col = 0
        for item in infantry_troops:
            button = Radiobutton(troops_type_label_frame, text=item, variable=infantry_troops_training_type_auto,
                                 value=item,
                                 indicatoron=False)
            button.grid(row=4, column=col)
            col += 1

        cavalry_troops_training_type_auto = StringVar()
        cavalry_troops = self.travian_controller.tribe.get_troops_by_type(troops_type=TroopType.CAVALRY)
        col = 0
        for item in cavalry_troops:
            button = Radiobutton(troops_type_label_frame, text=item, variable=cavalry_troops_training_type_auto,
                                 value=item,
                                 indicatoron=False)
            button.grid(row=5, column=col)
            col += 1

        auto_mode_farm_raid_frame = LabelFrame(auto_mode_label_frame, text="Farms Raid")
        Radiobutton(auto_mode_farm_raid_frame, text="Oasis", variable=farm_type_auto,
                    indicatoron=False, value=FarmType.OASIS.value, width=8).grid(
            row=5, column=0)
        Radiobutton(auto_mode_farm_raid_frame, text="Farm", variable=farm_type_auto,
                    indicatoron=False, value=FarmType.NORMAL_FARM.value, width=8).grid(
            row=5, column=1)

        auto_mode_troops_raid = StringVar()

        for item in troops:
            button = Radiobutton(auto_mode_farm_raid_frame, text=item, variable=auto_mode_troops_raid, value=item)
            button.grid(column=0)

        Label(auto_mode_farm_raid_frame, text="Number Of Soldiers").grid(row=10, column=5)
        number_of_troops_auto = Entry(auto_mode_farm_raid_frame, bd=5)
        number_of_troops_auto.grid(row=10, column=1)
        Button(auto_mode_label_frame, text="Start Auto Mode",
               command=partial(self.on_start_auto_mode_click, farm_type_auto, number_of_troops_auto, is_batch,
                               is_barracks,
                               is_stable, infantry_troops_training_type_auto, cavalry_troops_training_type_auto,
                               auto_mode_troops_raid)).grid(row=10,
                                                            column=5)
        auto_mode_farm_raid_frame.grid(row=5, columnspan=7, sticky='WE',
                                       padx=5, pady=5, ipadx=5, ipady=5)
        troops_type_label_frame.grid(row=4, columnspan=7, sticky='WE',
                                     padx=5, pady=5, ipadx=5, ipady=5)
        auto_mode_label_frame.grid(row=3, columnspan=7, sticky='WE',
                                   padx=5, pady=5, ipadx=5, ipady=5)

        farm_list_label_frame = LabelFrame(self, text="Farm List")
        troops_raid_type_label_frame = LabelFrame(farm_list_label_frame, text="Troops Raid Type")

        self.troops_raid = StringVar()

        for item in troops:
            button = Radiobutton(troops_raid_type_label_frame, text=item, variable=self.troops_raid, value=item)
            button.grid(column=0)
        Label(farm_list_label_frame, text="Number Of Soldiers").grid(row=2, column=0)
        self.number_of_troops_entry = Entry(farm_list_label_frame, bd=5)
        self.number_of_troops_entry.grid(row=2, column=2)

        self.farm_type = IntVar()
        self.farm_type_label_frame = LabelFrame(farm_list_label_frame, text="Farm Type")
        self.r1 = Radiobutton(self.farm_type_label_frame, text="Oasis", variable=self.farm_type,
                              indicatoron=False, value=FarmType.OASIS.value, width=8)
        self.r2 = Radiobutton(self.farm_type_label_frame, text="Farm", variable=self.farm_type,
                              indicatoron=False, value=FarmType.NORMAL_FARM.value, width=8)
        troops_raid_type_label_frame.grid(row=1, columnspan=6, sticky='WE',
                                          padx=5, pady=5, ipadx=5, ipady=5)
        farm_list_label_frame.grid(row=6, columnspan=7, sticky='WE',
                                   padx=5, pady=5, ipadx=5, ipady=5)

        self.send_btn = Button(self.farm_type_label_frame, text="Send",
                               command=partial(self.on_farm_raid_send_click, self.farm_type, self.troops_raid,
                                               self.number_of_troops_entry))
        farm_list_action = partial(self.on_add_farms_to_farm_list_click, self.troops_raid,
                                   self.number_of_troops_entry)
        Button(farm_list_label_frame, text="Farm account to farm list",
               command=farm_list_action).grid(row=6, column=0)
        Button(farm_list_label_frame, text="Farm raid",
               command=self.on_farm_raid_click).grid(row=6, column=2)

    def on_start_auto_mode_click(self, farm_type, number_of_troops, is_batch, is_barracks, is_stable,
                                 infantry_troops_training_type, cavalry_troops_training_type, troops_raiding_type):
        self.travian_controller.auto_mode(farm_type.get(), number_of_troops.get(), is_batch.get(), is_barracks.get(),
                                          is_stable.get(), infantry_troops_training_type.get(),
                                          cavalry_troops_training_type.get(), troops_raiding_type.get())

    def on_add_farms_to_farm_list_click(self, troops_type, number_of_troops_entry):
        self.travian_controller.add_farm_account_to_farm_list(troops_type.get(), number_of_troops_entry.get())
        self.r1.grid_forget()
        self.r2.grid_forget()
        self.farm_type_label_frame.grid_forget()

    def on_farm_raid_send_click(self, farm_type, troops_type, number_of_troops=None):
        self.travian_controller.raid_farms(farm_type.get(), troops_type.get(), number_of_troops.get())

    def on_farm_raid_click(self):
        self.r1.grid(
            row=6, column=6)
        self.r2.grid(
            row=6, column=7)
        self.farm_type_label_frame.grid(row=1, column=7, columnspan=6, sticky='WE',
                                        padx=5, pady=5, ipadx=5, ipady=5)
        self.send_btn.grid(row=7, column=7)

    def on_click(self, x, y, button, pressed):
        if pressed and button == Btn.middle:
            self.travian_controller.add_oasis_to_custom_farm_list()
        elif button == Btn.right:
            return False

    def unregister_click_listener(self):
        if self.listener is not None:
            self.listener.stop()

    def register_click_listener(self):
        self.listener = Listener(
            on_click=self.on_click)
        self.listener.start()


if __name__ == '__main__':
    main()