ticket/MainPanel.py at master · whrxiao/ticket · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
# coding=utf-8
import wx
import checker
import datetime
from checker import ticket_type, train_class
from MultiCheckBox import MultiCheckBox
from TimeSelector import TimeSelector
from DateSelector import DateSelector
from settings import info
import os


class MPanel(wx.Panel):

    """docstring for ClassName"""
    def __init__(self, parent):
        self.checking = False
        date = "Date"
        fromCity = 'From'
        toCity = 'To'
        time = 'Time'
        seat = 'Seat'
        email = 'Email'
        start = 'Start'
        mainSizer = wx.BoxSizer(wx.VERTICAL)

        cityBox = wx.BoxSizer(wx.HORIZONTAL)
        wx.Panel.__init__(self, parent)
        self.fromCityLabel = wx.StaticText(self, label=fromCity)
        self.fromCityInput = wx.TextCtrl(self, size=(140, -1))
        self.toCityLabel = wx.StaticText(self, label=toCity)
        self.toCityInput = wx.TextCtrl(self, size=(140, -1))
        cityBox.Add(self.fromCityLabel)
        cityBox.Add(self.fromCityInput)
        cityBox.Add(self.toCityLabel)
        cityBox.Add(self.toCityInput)

        dateBox = wx.BoxSizer(wx.HORIZONTAL)
        self.dateLabel = wx.StaticText(self, label=date)

        self.startDateCombo = DateSelector(self)
        self.endDateCombo = DateSelector(self)
        dateBox.AddMany([self.dateLabel, self.startDateCombo, self.endDateCombo])

        timeBox = wx.BoxSizer(wx.HORIZONTAL)
        self.timeLabel = wx.StaticText(self, label=time)
        self.timeInputFrom = TimeSelector(self)
        self.timeInputTo = TimeSelector(self)
        timeBox.Add(self.timeLabel)
        timeBox.Add(self.timeInputFrom)
        timeBox.Add(self.timeInputTo)

        self.classCheckBoxGroup = MultiCheckBox(self, train_class)
        self.typeCheckBoxGroup = MultiCheckBox(self, ticket_type)

        mailBox = wx.BoxSizer(wx.VERTICAL)
        self.mailLabel = wx.StaticText(self, label=email)
        self.mailInput = wx.TextCtrl(self, size=(140, -1))
        mailBox.Add(self.mailLabel)
        mailBox.Add(self.mailInput)

        self.rememberCheckbox = wx.CheckBox(self, label="Remember Settings")
        self.rememberCheckbox.SetValue(True)

        self.startButton = wx.Button(self, label=start)
        self.Bind(wx.EVT_BUTTON, self.OnStartClick, self.startButton)

        self.log = wx.TextCtrl(self, -1, "",
                               style=wx.TE_RICH | wx.TE_MULTILINE)

        if 'config' in os.listdir('.'):
            old_info = info()
            old_info.load()
            # print old_info.cities[0]
            self.fromCityInput.SetValue(old_info.cities[0])
            self.toCityInput.SetValue(old_info.cities[1])
            # print old_info.dates[0]
            self.startDateCombo.setValue(old_info.dates[0])
            self.endDateCombo.setValue(old_info.dates[1])
            self.timeInputFrom.setValue(old_info.time_limit[0])
            self.timeInputTo.setValue(old_info.time_limit[1])
            self.classCheckBoxGroup.setValue(old_info.train_c)
            self.typeCheckBoxGroup.setValue(old_info.ticket_t)
            self.mailInput.SetValue(old_info.email)

        mainSizer.Add(cityBox, 0, wx.ALL, 5)
        mainSizer.Add(dateBox, 0, wx.ALL, 5)
        mainSizer.Add(timeBox, 0, wx.ALL, 5)
        mainSizer.Add(self.classCheckBoxGroup, 0, wx.ALL, 5)
        mainSizer.Add(self.typeCheckBoxGroup, 0, wx.ALL, 5)
        mainSizer.Add(mailBox, 0, wx.ALL, 5)
        mainSizer.Add(self.rememberCheckbox, 0, wx.ALL, 5)
        mainSizer.Add(self.startButton, 0, wx.ALL, 5)
        mainSizer.Add(self.log, 1, wx.EXPAND | wx.ALL, 5)

        self.SetSizerAndFit(mainSizer)

    def OnStartClick(self, e):

        if self.checking:
            self.workerThread.stop()
            self.checking = False
            return
        print self.mailInput.GetValue()

        cities = []
        cities.append(self.fromCityInput.GetValue())
        cities.append(self.toCityInput.GetValue())
        time_limit = [self.timeInputFrom.getValue(), self.timeInputTo.getValue()]
        email = self.mailInput.GetValue()
        ticket_t = self.typeCheckBoxGroup.getValue()
        train_c = self.classCheckBoxGroup.getValue()
        print train_c
        print self.startDateCombo.getValue().isoformat(), self.endDateCombo.getValue().isoformat()

        startDay = self.startDateCombo.getValue()
        endDay = self.endDateCombo.getValue()
        s_info = info([startDay.isoformat(), endDay.isoformat()], time_limit, cities, ticket_t, train_c, email)
        if self.rememberCheckbox.IsChecked():
            s_info.save()

        self.workerThread = checker.Checker(s_info, self)
        self.workerThread.start()
        self.checking = True

    def logging(self, msg):
        self.log.AppendText(msg+'\n')