changeset 2:e0061735c327

Usunąłem nieskończoną pętlę
author michalr
date Tue, 22 Feb 2011 18:37:32 +0000
parents c3fb1e9fc1f7
children 79fc37fe85a6
files frontend/mfrontend/__init__.py frontend/mfrontend/utils.py
diffstat 2 files changed, 16 insertions(+), 8 deletions(-) [+]
line wrap: on
line diff
--- a/frontend/mfrontend/__init__.py	Tue Feb 22 18:10:18 2011 +0000
+++ b/frontend/mfrontend/__init__.py	Tue Feb 22 18:37:32 2011 +0000
@@ -2,21 +2,21 @@
 # -*- coding: utf-8 -*-
 
 import flask
-import ConfigParser
+from ConfigParser import SafeConfigParser, NoSectionError
 from mfrontend.views.frontend import frontend
 from mfrontend import db, utils
 
 
 #-- konfiguracja
-config = ConfigParser.SafeConfigParser()
-config.read('config.ini')
+config = SafeConfigParser()
 while True:
     try:
+        config.read('config.ini')
         DATABASE = config.get('Basic', 'Database')
         DEBUG = config.getboolean('Basic', 'Debug')
         SECRET_KEY = config.get('Basic', 'Secret_Key')
         break
-    except ConfigParser.NoSectionError:
+    except NoSectionError:
         utils.default_ini()
 #-- koniec konfiguracji
 
--- a/frontend/mfrontend/utils.py	Tue Feb 22 18:10:18 2011 +0000
+++ b/frontend/mfrontend/utils.py	Tue Feb 22 18:37:32 2011 +0000
@@ -1,16 +1,24 @@
 #!/usr/bin/python
 # -*- coding: utf-8 -*-
 
-import random
+import random, string
+from ConfigParser import SafeConfigParser
+
+def randomString(n):
+    """Funkcja generująca losowy łańcuch znaków, np. do haseł czy Secret Key.
+       @param n Ilość znaków do wygenerowania.
+       @return Losowo wygenerowany łańcuch znaków.
+    """
+    wynik = ''.join(random.choice(string.ascii_letters + string.digits) for i in range(n))
+    return wynik
 
 def default_ini():
     """Funkcja tworząca domyślny plik .ini z ustawieniami aplikacji"""
-    config = ConfigParser.SafeConfigParser()
+    config = SafeConfigParser()
     config.add_section('Basic')
     config.set('Basic', 'Database', '../baza.sqlite')
     config.set('Basic', 'Debug', '0')
-    config.set('Basic', 'Secret_Key',
-               ''.join(random.choice(string.ascii_letters for x in range(10))))
+    config.set('Basic', 'Secret_Key', randomString(10))
     with open('config.ini', 'wb') as configfile:
         config.write(configfile)