From 80ef752a52efb9c73c9559ad5516335ec0bd408d Mon Sep 17 00:00:00 2001
From: Amjith Ramanujam <amjith@newrelic.com>
Date: Thu, 20 Nov 2014 23:15:50 -0800
Subject: [PATCH] Split the classes to separate files and upgrade
 prompt_toolkit.

---
 pgcli.py       | 66 +++++++++++++-------------------------------------
 pgcompleter.py | 20 +++++++++++++++
 pgstyle.py     | 15 ++++++++++++
 3 files changed, 52 insertions(+), 49 deletions(-)
 create mode 100644 pgcompleter.py
 create mode 100644 pgstyle.py

diff --git a/pgcli.py b/pgcli.py
index b8f19f3b..7b21332b 100755
--- a/pgcli.py
+++ b/pgcli.py
@@ -1,69 +1,37 @@
 #!/usr/bin/env python
 from __future__ import unicode_literals
-import sys
+from __future__ import print_function
 
 import click
 import psycopg2
 
 from prompt_toolkit import CommandLineInterface, AbortAction, Exit
-from prompt_toolkit.completion import Completer, Completion
+from pgcompleter import PGCompleter
+from pgstyle import PGStyle
 from prompt_toolkit.line import Line
 from prompt_toolkit.layout import Layout
 from prompt_toolkit.layout.prompt import DefaultPrompt
-from prompt_toolkit.layout.menus import CompletionMenu
-
-from pygments.token import Token
-from pygments.style import Style
-from pygments.lexers.sql import SqlLexer
-from pygments.styles.default import DefaultStyle
-
-
-class SqlCompleter(Completer):
-    keywords = [
-        'SELECT',
-        'INSERT',
-        'ALTER',
-        'DROP',
-        'DELETE',
-        'FROM',
-        'WHERE',
-    ]
-
-    def get_completions(self, document):
-        word_before_cursor = document.get_word_before_cursor()
-
-        for keyword in self.keywords:
-            if (keyword.startswith(word_before_cursor) or
-                    keyword.startswith(word_before_cursor.upper())):
-                yield Completion(keyword, -len(word_before_cursor))
-
-
-class DocumentStyle(Style):
-    styles = {
-            Token.Menu.Completions.Completion.Current: 'bg:#00aaaa #000000',
-            Token.Menu.Completions.Completion: 'bg:#008888 #ffffff',
-            Token.Menu.Completions.ProgressButton: 'bg:#003333',
-            Token.Menu.Completions.ProgressBar: 'bg:#00aaaa',
-            Token.SelectedText: '#ffffff bg:#6666aa',
-            Token.IncrementalSearchMatch: '#ffffff bg:#4444aa',
-            Token.IncrementalSearchMatch.Current: '#ffffff bg:#44aa44',
-            }
-    styles.update(DefaultStyle.styles)
+from prompt_toolkit.layout.menus import CompletionsMenu
 
+from pygments.lexers import SqlLexer
 
 @click.command()
-def pgcli():
+@click.option('-h', '--host', default='localhost')
+@click.option('-U', '--user', prompt=True, envvar='USER')
+@click.password_option('-P', '--password', default='',
+        confirmation_prompt=False)
+@click.argument('database', envvar='USER')
+def pgcli(database, host, user, password):
+    print('database: %s host: %s user: %s password: %s' % (database, host, user, password))
     layout = Layout(before_input=DefaultPrompt('> '),
-            menus=[CompletionMenu()],
+            menus=[CompletionsMenu()],
             lexer=SqlLexer)
-    line = Line(completer=SqlCompleter())
-    cli = CommandLineInterface(style=DocumentStyle, layout=layout, line=line)
+    line = Line(completer=PGCompleter())
+    cli = CommandLineInterface(style=PGStyle, layout=layout, line=line)
 
     try:
         while True:
             document = cli.read_input(on_exit=AbortAction.RAISE_EXCEPTION)
-            print 'You entered:', document.text
+            print ('You entered:', document.text)
     except Exit:
-        print 'GoodBye!'
-
-
+        print ('GoodBye!')
diff --git a/pgcompleter.py b/pgcompleter.py
new file mode 100644
index 00000000..f8c0338b
--- /dev/null
+++ b/pgcompleter.py
@@ -0,0 +1,20 @@
+from prompt_toolkit.completion import Completer, Completion
+
+class PGCompleter(Completer):
+    keywords = [
+        'SELECT',
+        'INSERT',
+        'ALTER',
+        'DROP',
+        'DELETE',
+        'FROM',
+        'WHERE',
+    ]
+
+    def get_completions(self, document):
+        word_before_cursor = document.get_word_before_cursor()
+
+        for keyword in self.keywords:
+            if (keyword.startswith(word_before_cursor) or
+                    keyword.startswith(word_before_cursor.upper())):
+                yield Completion(keyword, -len(word_before_cursor))
diff --git a/pgstyle.py b/pgstyle.py
new file mode 100644
index 00000000..9cbb73b2
--- /dev/null
+++ b/pgstyle.py
@@ -0,0 +1,15 @@
+from pygments.token import Token
+from pygments.style import Style
+from pygments.styles.default import DefaultStyle
+
+class PGStyle(Style):
+    styles = {
+            Token.Menu.Completions.Completion.Current: 'bg:#00aaaa #000000',
+            Token.Menu.Completions.Completion: 'bg:#008888 #ffffff',
+            Token.Menu.Completions.ProgressButton: 'bg:#003333',
+            Token.Menu.Completions.ProgressBar: 'bg:#00aaaa',
+            Token.SelectedText: '#ffffff bg:#6666aa',
+            Token.IncrementalSearchMatch: '#ffffff bg:#4444aa',
+            Token.IncrementalSearchMatch.Current: '#ffffff bg:#44aa44',
+            }
+    styles.update(DefaultStyle.styles)