better error handling

Author: 3on

app/app.py

@@ -24,12 +24,17 @@
 
 # PostgreSQL connection
 import psycopg2
-conn = psycopg2.connect(database="test",
-                        user=dotcloud_env['DOTCLOUD_DB_SQL_LOGIN'],
-                        password=dotcloud_env['DOTCLOUD_DB_SQL_PASSWORD'],
-                        host=dotcloud_env['DOTCLOUD_DB_SQL_HOST'],
-                        port=int(dotcloud_env['DOTCLOUD_DB_SQL_PORT']))
-cur = conn.cursor()
+
+try:
+    conn = psycopg2.connect(database="test",
+                            user=dotcloud_env['DOTCLOUD_DB_SQL_LOGIN'],
+                            password=dotcloud_env['DOTCLOUD_DB_SQL_PASSWORD'],
+                            host=dotcloud_env['DOTCLOUD_DB_SQL_HOST'],
+                            port=int(dotcloud_env['DOTCLOUD_DB_SQL_PORT']))
+    cur = conn.cursor()
+except Exception as e:
+    print e
+    exit(1)
 
 @app.route("/")
 def hello():

app/postgresql.py

@@ -4,6 +4,7 @@
 from os import path
 from time import sleep
 from sys import stdout as out
+import sys
 
 dotcloud_env_file_path = path.expanduser('~/environment.json')
 if path.exists(dotcloud_env_file_path):
@@ -21,7 +22,8 @@
 
 out.write("Creating the database...")
 out.flush()
-for i in xrange(1,30):
+i = 60
+while True:
     try:
         conn = psycopg2.connect(
                                 user=env['DOTCLOUD_DB_SQL_LOGIN'],
@@ -36,6 +38,9 @@
         out.flush()
         break
     except Exception as e:
+        i -= 1
+        if i <= 0 :
+            sys.exit(1)
         sleep(1)
         out.write(".")
         out.flush()