Start rewrite

main.py

@@ -0,0 +1,97 @@
+#!/usr/bin/env python3
+
+import json
+import os
+
+import asana
+from asana.session import AsanaOAuth2Session
+import urwid
+
+import secrets
+from ui.auth import AuthPrompt
+from ui.palette import palette
+
+class NotAuthedException(Exception):
+    def __init__(self):
+        super(NotAuthedException, self)
+
+class CmdAsana(object):
+    """The main urwid loop
+    """
+    loop = None
+
+    """Try to get an Asana client using stored tokens
+
+    Raises:
+        NotAuthedException: the user has not authorized the app
+    """
+    def get_client(self):
+        try:
+            f = open('.oauth', 'r')
+            auth_json = f.read()
+            f.close()
+            token = json.loads(auth_json)
+            self.client = asana.Client.oauth(
+                client_id=secrets.CLIENT_ID,
+                client_secret=secrets.CLIENT_SECRET,
+                token=token,
+                token_updater=self.save_token,
+                auto_refresh_url=AsanaOAuth2Session.token_url,
+            )
+        except IOError:
+            raise NotAuthedException()
+
+    def authorize(self):
+        self.client = asana.Client.oauth(
+            client_id=secrets.CLIENT_ID,
+            client_secret=secrets.CLIENT_SECRET,
+            redirect_uri='urn:ietf:wg:oauth:2.0:oob',
+            token_updater=self.save_token,
+            auto_refresh_url=AsanaOAuth2Session.token_url,
+        )
+        (url, _) = self.client.session.authorization_url()
+        auth = AuthPrompt(url, self.auth_callback)
+
+        try:
+            import webbrowser
+            webbrowser.open(url)
+        except Exception:
+            pass
+
+        self.loop = urwid.MainLoop(
+            auth.component(),
+            unhandled_input=self.exit_handler,
+            palette=palette
+        )
+        self.loop.run()
+
+    def auth_callback(self, code):
+        self.save_token(
+            self.client.session.fetch_token(code=code))
+        raise urwid.ExitMainLoop()
+
+
+    def exit_handler(self, key):
+        if key in ('q', 'Q', 'esc'):
+            raise urwid.ExitMainLoop()
+
+    def run(self):
+        print("Running...")
+
+    def save_token(self, token):
+        f = open('.oauth', 'w')
+        f.write(json.dumps(token))
+        f.close()
+        os.chmod('.oauth', 0o600)
+
+def main():
+    cmd_asana = CmdAsana()
+    try:
+        cmd_asana.get_client()
+    except NotAuthedException:
+        cmd_asana.authorize()
+
+    cmd_asana.run()
+
+if __name__ == "__main__":
+    main()