Add gui

diff --git a/gui.py b/gui.py
new file mode 100644 (file)
index 0000000..6e4969e
--- /dev/null
+++ b/gui.py
@@ -0,0 +1,185 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+from tkinter import Tk, Frame, Label, Entry, StringVar, Button, IntVar, Checkbutton, filedialog
+import asyncio
+from getmyancestors import Session, Tree
+
+
+class LogVar(StringVar):
+
+    def write(self, text):
+        self.set(self.get() + text)
+
+
+class SignIn(Frame):
+
+    def __init__(self, window, **kwargs):
+        super(SignIn, self).__init__(window, **kwargs)
+        self.username = StringVar()
+        self.password = StringVar()
+        label_username = Label(self, text='Username')
+        entry_username = Entry(self, textvariable=self.username, width=30)
+        label_password = Label(self, text='Password')
+        entry_password = Entry(self, show='●', textvariable=self.password, width=30)
+        label_username.pack()
+        entry_username.pack()
+        label_password.pack()
+        entry_password.pack()
+
+
+class StartIndi(Frame):
+
+    def __init__(self, window, **kwargs):
+        super(StartIndi, self).__init__(window, **kwargs)
+        self.label_fid = Label(self, text='')
+        self.fid = StringVar()
+        self.btn_delete = Button(self, text='delete', command=self.delete)
+        entry_fid = Entry(self, textvariable=self.fid, width=10)
+        self.label_fid.pack()
+        entry_fid.pack()
+        self.btn_delete.pack()
+
+    def delete(self):
+        self.master.master.start_indis.remove(self)
+        self.destroy()
+
+
+class Options(Frame):
+    def __init__(self, window, ordinances=False, **kwargs):
+        super(Options, self).__init__(window, **kwargs)
+        self.ancestors = IntVar()
+        self.ancestors.set(4)
+        self.descendants = IntVar()
+        self.spouses = IntVar()
+        self.ordinances = IntVar()
+        self.contributors = IntVar()
+        self.start_indis = list()
+        self.indis = Frame(self)
+        self.filename = None
+        label_ancestors = Label(self, text='Number of generations to ascend')
+        entry_ancestors = Entry(self, textvariable=self.ancestors, width=3)
+        label_descendants = Label(self, text='Number of generations to descend')
+        entry_descendants = Entry(self, textvariable=self.descendants, width=3)
+        btn_add_indi = Button(self, text='add', command=self.add_indi)
+        btn_spouses = Checkbutton(self, text='Add spouses and couples information', variable=self.spouses)
+        btn_ordinances = Checkbutton(self, text='Add temple information', variable=self.ordinances)
+        btn_contributors = Checkbutton(self, text='Add list of contributors in notes', variable=self.contributors)
+        askfilename = Button(self, text='Save as', command=self.askfilename)
+        label_ancestors.pack()
+        entry_ancestors.pack()
+        label_descendants.pack()
+        entry_descendants.pack()
+        self.indis.pack()
+        btn_add_indi.pack()
+        btn_spouses.pack()
+        if ordinances:
+            btn_ordinances.pack()
+        btn_contributors.pack()
+        askfilename.pack()
+
+    def askfilename(self):
+        self.filename = filedialog.asksaveasfilename()
+
+    def add_indi(self, data=None):
+        new_indi = StartIndi(self.indis)
+        self.start_indis.append(new_indi)
+        if data and 'persons' in data:
+            indi = data['persons'][0]
+            new_indi.fid.set(indi['id'])
+            if 'names' in data:
+                for name in data['names']:
+                    if name['preferred']:
+                        new_indi.label_fid['text'] = name['nameForms'][0]['fullText']
+                        break
+        new_indi.pack()
+
+
+class Gui(Frame):
+    def __init__(self, window, **kwargs):
+        super(Gui, self).__init__(window, borderwidth=10, **kwargs)
+        self.fs = None
+        self.tree = None
+        self.logfile = LogVar()
+        self.form = Frame(self)
+        self.sign_in = SignIn(self.form)
+        self.options = Options(self.form, True)
+        self.title = Label(self, text='Sign In to FamilySearch')
+        self.btn_quit = Button(self, text='Quit', command=self.quit)
+        self.btn_valid = Button(self, text='Sign In', fg='red', command=self.login)
+        self.title.pack()
+        self.sign_in.pack()
+        self.form.pack()
+        self.btn_quit.pack(side='left')
+        self.btn_valid.pack(side='right')
+        self.pack()
+
+    def login(self):
+        self.fs = Session(self.sign_in.username.get(), self.sign_in.password.get(), verbose=True, timeout=1)
+        if not self.fs.logged:
+            self.logfile.write('Connection failed')
+            return
+        self.tree = Tree(self.fs)
+        data = self.fs.get_url('/platform/tree/persons/%s.json' % self.fs.get_userid())
+        self.options.add_indi(data)
+        self.sign_in.destroy()
+        self.title['text'] = 'Options'
+        self.btn_valid['text'] = 'Download gedcom file'
+        self.btn_valid['command'] = self.download
+        self.options.pack()
+
+    def download(self):
+        if not self.options.filename:
+            print('Please choose a path')
+            return
+        todo = [start_indi.fid.get() for start_indi in self.options.start_indis]
+        self.tree.add_indis(todo)
+        todo = set(todo)
+        done = set()
+        for i in range(self.options.ancestors.get()):
+            if not todo:
+                break
+            done |= todo
+            todo = self.tree.add_parents(todo) - done
+
+        todo = set(self.tree.indi.keys())
+        done = set()
+        for i in range(self.options.descendants.get()):
+            if not todo:
+                break
+            done |= todo
+            todo = self.tree.add_children(todo) - done
+
+        if self.options.spouses.get():
+            todo = set(self.tree.indi.keys())
+            self.tree.add_spouses(todo)
+
+        async def download_stuff(loop):
+            futures = set()
+            for fid, indi in self.tree.indi.items():
+                futures.add(loop.run_in_executor(None, indi.get_notes))
+                if self.options.ordinances.get():
+                    futures.add(loop.run_in_executor(None, self.tree.add_ordinances, fid))
+                if self.options.contributors.get():
+                    futures.add(loop.run_in_executor(None, indi.get_contributors))
+            for fam in self.tree.fam.values():
+                futures.add(loop.run_in_executor(None, fam.get_notes))
+                if self.options.contributors.get():
+                    futures.add(loop.run_in_executor(None, fam.get_contributors))
+            for future in futures:
+                await future
+
+        loop = asyncio.get_event_loop()
+        loop.run_until_complete(download_stuff(loop))
+
+        self.tree.reset_num()
+        file = open(self.options.filename, 'w')
+        self.tree.print(file)
+        file.close()
+        self.options.filename = None
+
+
+window = Tk()
+sign_in = Gui(window)
+
+sign_in.mainloop()