1 files changed, 103 insertions, 0 deletions

diff --git a/quoins/command.py b/quoins/command.py
new file mode 100644
index 0000000..2e6411a
--- /dev/null
+++ b/quoins/command.py
@@ -0,0 +1,103 @@
+import os
+from paste.script import command
+from paste.deploy import appconfig
+
+def get_config(self, config_spec):
+    section = self.options.section_name
+    if section is None:
+        if '#' in config_spec:
+            config_spec, section = config_spec.split('#', 1)
+        else:
+            section = 'main'
+    if not ':' in section:
+        plain_section = section
+        section = 'app:'+section
+    else:
+        plain_section = section.split(':', 1)[0]
+    if not config_spec.startswith('config:'):
+        config_spec = 'config:' + config_spec
+    if plain_section != 'main':
+        config_spec += '#' + plain_section
+    config_file = config_spec[len('config:'):].split('#', 1)[0]
+    config_file = os.path.join(os.getcwd(), config_file)
+    self.logging_file_config(config_file)
+    conf = appconfig(config_spec, relative_to=os.getcwd())
+    return conf
+
+class OpenIDCommand(command.Command):
+    max_args = 1
+    min_args = 1
+
+    usage = "CONFIG_FILE"
+    summary = "Setup Quoins OpenID tables"
+    group_name = "Quoins"
+
+    parser = command.Command.standard_parser(verbose=True)
+    parser.add_option('--name',
+                      action='store',
+                      dest='section_name',
+                      default=None,
+                      help='The name of the section to set up (default: app:main)')
+
+    def command(self):
+        config_file = self.args[0]
+        if self.verbose:
+            print "Using config file: %s" % config_file
+
+        conf = get_config(self, config_file)
+        
+        from quoins.controllers import get_oid_connection
+        from openid.store.sqlstore import MySQLStore
+        import MySQLdb
+
+        conn = get_oid_connection(conf)
+        store = MySQLStore(conn)
+        try:
+            store.createTables()
+        except MySQLdb.OperationalError, message:
+            errorcode = message[0]
+            if errorcode == 1050:
+                print 'ok'
+            else:
+                raise
+
+class BlogCommand(command.Command):
+    max_args = 1
+    min_args = 1
+
+    usage = "CONFIG_FILE"
+    summary = "Create a new Quoins Blog"
+    group_name = "Quoins"
+
+    parser = command.Command.standard_parser(verbose=True)
+    parser.add_option('--name',
+                      action='store',
+                      dest='section_name',
+                      default=None,
+                      help='The name of the section to set up (default: app:main)')
+
+    def command(self):
+        config_file = self.args[0]
+        if self.verbose:
+            print "Using config file: %s" % config_file
+
+        conf = get_config(self, config_file)
+        
+        from model import Blog, DBSession, init_model
+        from sqlalchemy import create_engine
+        init_model(create_engine(conf.get('sqlalchemy.url')))
+
+        title = raw_input("Blog title: ")
+        subtitle = raw_input("Blog subtitle: ")
+        comments = raw_input("Allow comments by default? (y/n) ")
+        if comments.strip().lower()=='y':
+            comments = True
+        else:
+            comments = False
+        b = Blog()
+        b.title = title.strip()
+        if subtitle:
+            b.subtitle = subtitle.strip()
+        b.allow_comments = comments
+        DBSession.add(b)
+        DBSession.flush()