Ajout support serveur MAIL et XMPP

tools/databaseinit.py

@@ -1,50 +1,72 @@
-#!venv/bin/python
-
+from flask import Flask
 import sqlite3
-import os.path
+import os
+
+app = Flask( 'pywallter' )
+app.config.from_pyfile('config.py')
+
+
+DATABASE = app.config['DATABASE']
+DOSSIER_PERSO = app.config['DOSSIER_APP']
+DATABASE = app.config['DATABASE']
+
+
 
 def init_db():
-    if os.path.isfile('base.db'):
-        return False
-    else:
-        conn = sqlite3.connect('base.db')
-        cursor = conn.cursor()
-        cursor.execute("""
-            CREATE TABLE IF NOT EXISTS users(
-            Mail CHAR(80) PRIMARY KEY UNIQUE NOT NULL,
-            name TEXT,
-            alias TEXT,
-            xmpp TEXT,
-            passwd TEXT,
-            avatar TEXT,
-            nom TEXT,
-            prenom TEXT,
-            age TEXT,
-            website TEXT,
-            Token CHAR(30),
-            Mail_rescue TEXT )
-        """)
-        conn.commit()
-        print ('table users OK')
+    conn = sqlite3.connect(DATABASE)
+    cursor = conn.cursor()
+    cursor.execute("""
+    CREATE TABLE IF NOT EXISTS users(
+    Mail CHAR(80) PRIMARY KEY UNIQUE NOT NULL,
+    name TEXT,
+    alias TEXT,
+    xmpp TEXT,
+    passwd TEXT,
+    avatar TEXT,
+    nom TEXT,
+    prenom TEXT,
+    age TEXT,
+    website TEXT,
+    Token CHAR(30),
+    invitations INTEGER DEFAULT (20),
+    Mail_rescue TEXT )
+    """)
+    conn.commit()
+    print ('table users Ok')
 
-        cursor.execute("""
-            CREATE TABLE IF NOT EXISTS posts(
-            id INTEGER PRIMARY KEY AUTOINCREMENT UNIQUE,
-            title TEXT,
-            content TEXT,
-            time TEXT,
-            category TEXT,
-            author TEXT,
-            status TEXT
-            )
-        """)
-        conn.commit()
-        conn.close()
-        print ('table posts OK')
-        return True
+    cursor.execute("""
+    CREATE TABLE IF NOT EXISTS posts(
+    id INTEGER PRIMARY KEY AUTOINCREMENT UNIQUE,
+    title TEXT,
+    content TEXT,
+    time TEXT,
+    category TEXT,
+    author TEXT,
+    status TEXT
+    )
+    """)
+    conn.commit()
+    conn.close()
+    print ('table posts OK')
 
 def init_dir():
     if os.path.isdir('users'):
         return False
     else:
         os.makedirs('./users/')
+
+def db_migrate():
+    conn = sqlite3.connect(DATABASE)
+    cursor = conn.cursor()
+
+    cursor.execute("""SELECT name FROM PRAGMA_TABLE_INFO('users');""")
+    db_columns = cursor.fetchall()
+    present = False
+    for col in db_columns:
+        if "invitations" == col[0]:
+            present = True
+
+    if not(present):
+        cursor.execute("""ALTER TABLE users ADD COLUMN invitations INTEGER DEFAULT (20);""")
+        conn.commit()
+        print ("Ajout du champ invitations")