EDProxy/edimport.py at master · Imagix/EDProxy · 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
import wx
import edconfig
import edutils
import edevent

db_path_selector = {
    0: "C:\\Program Files (x86)\\EDDiscovery",
    wx.NOT_FOUND: "C:\\Program Files (x86)\\EDDiscovery"
}

class EDImportDialog(wx.Dialog):
    def __init__(self, *args, **kwargs):
        wx.Dialog.__init__(self, *args, **kwargs)

        self._edconfig = edconfig.get_instance()

        self._import_path = wx.TextCtrl(self)
        self._import_browse_button = wx.Button(self, wx.ID_ANY, _("Browse"))

        self._import_db = wx.Choice(self, choices = [ "EDDiscovery" ])

        self._import_path.SetMinSize((467, 29))
        self._import_path.ChangeValue(db_path_selector[self._import_db.GetSelection()])

#         self.Bind(wx.EVT_BUTTON, self.__on_ok, id=wx.ID_OK)
        self.Bind(wx.EVT_BUTTON, self.__on_import_browse, self._import_browse_button)

        self.__do_layout()

    def __do_layout(self):
        box1 = wx.BoxSizer(wx.VERTICAL)
        box2 = wx.BoxSizer(wx.HORIZONTAL)

        button_sizer = self.CreateSeparatedButtonSizer(wx.CANCEL | wx.OK)

        # Add all the main panel boxes to the top-level sizer
        box1.AddSpacer(10)
        box1.Add(box2, 0, wx.EXPAND)
        box1.Add(button_sizer, 0, wx.EXPAND | wx.ALIGN_RIGHT)

        box2.Add(self._import_db, 0, flag = wx.ALIGN_LEFT | wx.ALIGN_CENTER_HORIZONTAL | wx.ALIGN_CENTER_VERTICAL)
        box2.AddSpacer(5)
        box2.Add(self._import_path, 1)
        box2.AddSpacer(5)
        box2.Add(self._import_browse_button, flag = wx.ALIGN_RIGHT | wx.ALIGN_CENTER_HORIZONTAL | wx.ALIGN_CENTER_VERTICAL)

        self.SetSizer(box1)
        box1.Fit(self)
        self.Layout()
        self.Centre()

#     def __on_ok(self, event):
#         event.Skip()

    def __on_import_browse(self, event):
        defpath = self._import_path.GetValue()

        if len(defpath) == 0:
            defpath = edutils.get_user_dir()

        dir_path = wx.DirDialog(self, "Choose Netlog Path", style = wx.DD_DEFAULT_STYLE | wx.DD_DIR_MUST_EXIST, defaultPath = defpath)

        if dir_path.ShowModal() == wx.ID_OK:
            self._import_path.ChangeValue(dir_path.GetPath())

        dir_path.Destroy()
        event.Skip()

class _EDImportEvent(edevent.BaseEvent):
    def __init__(self,
                 date,
                 system_name,
                 position = (0.0, 0.0, 0.0),
                 main_star = "NA",
                 stellar_bodies = 1,
                 stellar_types = [ "NA" ],
                 planetary_bodies = 0,
                 planetary_types = [ "NA" ],
                 notes = "",
                 ref_distances = []):
        edevent.BaseEvent.__init__(self, "Import", date)

        self._system = system_name
        self._position = position
        self._main_star = main_star
        self._stellar_bodies = stellar_bodies
        self._stellar_types = stellar_types
        self._planetary_bodies = planetary_bodies
        self._planetary_types = planetary_types
        self._notes = notes
        self._ref_distances = ref_distances

    def _fill_json_dict(self, json_dict):
        json_dict['System'] = self._system
        json_dict['Position'] = self._position
        json_dict['MainStar'] = self._main_star
        json_dict['StellarBodies'] = self._stellar_bodies
        json_dict['StellarTypes'] = self._stellar_types
        json_dict['PlanetaryBodies'] = self._planetary_bodies
        json_dict['PlanetaryTypes'] = self._planetary_types
        json_dict['Notes'] = self._notes
        json_dict['RefDistances'] = self._ref_distances