colourmodes

2 files changed, 13 insertions, 6 deletions

diff --git a/alot.py b/alot.py
index 69c2697d..fd1cc351 100755
--- a/alot.py
+++ b/alot.py
@@ -8,6 +8,11 @@ from alot.ui import UI
 
 def parse_args():
     parser = argparse.ArgumentParser()
+    parser.add_argument('-c', dest='colours',
+                        type=int,
+                        choices=[1,16,88,256],
+                        default=16,
+                        help='colour mode')
     parser.add_argument('-r', dest='read_only',
                         action='store_true',
                         help='open db in read only mode')
@@ -15,7 +20,8 @@ def parse_args():
                         help='path to notmuch index')
     parser.add_argument('-d', dest='debug_level',
                         default='info',
-                        help='one of DEBUG,INFO,WARNING,ERROR')
+                        choices=['DEBUG','INFO','WARNING','ERROR'],
+                        help='debug level')
     parser.add_argument('-l', dest='logfile',
                         default='debug.log',
                         help='logfile')
@@ -33,7 +39,9 @@ def main():
     logger = logging.getLogger()
     ui = UI(db=dbman,
             log=logger,
-            initialquery=args.query)
+            initialquery=args.query,
+            colourmode = args.colours
+    )
 
 if __name__ == "__main__":
     main()
diff --git a/alot/ui.py b/alot/ui.py
index 7a094b4f..04ea8b65 100644
--- a/alot/ui.py
+++ b/alot/ui.py
@@ -10,18 +10,17 @@ class UI:
     buffers = []
     current_buffer = None
 
-    def __init__(self, db, log, initialquery):
+    def __init__(self, db, log, initialquery, colourmode):
         self.logger = log
         self.dbman = db
 
-        self.logger.debug('setup gui')
+        self.logger.debug('setup gui: %d'%colourmode)
         self.mainframe = urwid.Frame(urwid.SolidFill(' '))
         self.mainloop = urwid.MainLoop(self.mainframe,
                 settings.palette,
                 handle_mouse=False,
                 unhandled_input=self.keypress)
-        #self.mainloop.screen.set_terminal_properties(colors=256)
-        self.mainloop.screen.set_terminal_properties(colors=16)
+        self.mainloop.screen.set_terminal_properties(colors=colourmode)
 
         self.logger.debug('setup bindings')
         self.bindings = {'i': ('search', {'query': 'tag:inbox'}),