def insert_infected():
user_key = request.json['key']
db = config_db()
db.establish_connection()
query = "INSERT INTO infected VALUES (%s)"
params = user_key,
db.execute_query(query, params)
db.close_connection()
return "Success"
def add_new_user():
user_key = request.json['key']
db = config_db()
db.establish_connection()
query = "INSERT INTO curr_keys VALUES (%s, %s)"
params = user_key, None
db.execute_query(query, params)
db.close_connection()
return "Success"
def update_keys():
new_key = request.json['new_key']
old_key = request.json['old_key']
db = config_db()
db.establish_connection()
query = "UPDATE curr_keys SET curr_key = %s WHERE curr_key = %s"
params = new_key, old_key
db.execute_query(query, params)
db.close_connection()
return "Success"
def remove_user():
user_key = request.json['key']
db = config_db()
db.establish_connection()
params = user_key,
db.execute_query("DELETE FROM pairs WHERE dev1 = %s", params)
db.execute_query("DELETE FROM pairs WHERE dev2 = %s", params)
db.execute_query("DELETE FROM infected WHERE key = %s", params)
db.execute_query("DELETE FROM curr_keys WHERE curr_key = %s", params)
db.close_connection()
return "Success"
def add_new_pair():
user_one = request.json['key1']
user_two = request.json['key2']
db = config_db()
db.establish_connection()
time = datetime.now(timezone.utc)
if user_one > user_two:
query = "INSERT INTO pairs VALUES (%s, %s, %s)"
params = user_one, user_two, time
db.execute_query(query, params)
db.close_connection()
return "Success"
def user_status():
user_key = request.json['key']
db = config_db()
db.establish_connection()
cur = db.connection.cursor()
params = user_key, user_key
cur.execute("SELECT * FROM pairs WHERE dev1 = %s OR dev2 = %s", params)
instances = cur.fetchall()
for i in instances:
if i[3] or i[4]:
return "Positive"
cur.close()
db.close_connection()
return "Negative"
def get_complete_db_records(sql_query):
conn = None
try:
params = config_db()
conn = psycopg2.connect(**params)
cursor = conn.cursor()
cursor.execute(sql_query)
complete_db_records = cursor.fetchall()
cursor.close()
conn.commit()
return complete_db_records
except (Exception, psycopg2.DatabaseError) as error:
print(error)
finally:
if conn is not None:
conn.close()
import json
import sys
import psycopg2
import database
from config import config_db
if __name__ == '__main__':
if len(sys.argv) != 2:
raise Exception("This script is supposed to take a file as argument.")
with open(sys.argv[1], 'r') as file:
data = file.read()
db = config_db()
db.establish_connection()
db.execute_query(data)
print("Successfully read in PostgreSQL from file.")
db.close_connection()
def setup_db(app, database_path):
config_db(app, database_path)
db.app = app
db.init_app(app)
db.create_all()
migrate = Migrate(app, db)
# import config
import config
# Create cursor.
mydb = config.config_db()
mycursor = mydb.cursor()
#for creating ids
def createdata(firstname, lastname, email, phone, dob, gender, password):
sql = 'insert into userdata(firstname,lastname,email,phone,dob,\
gender,password) values(%s,%s,%s,%s,%s,%s,%s);'
val = (firstname, lastname, email, phone, dob, gender, password)
mycursor.execute(sql, val)
mydb.commit()
print(mycursor.rowcount, "record inserted.")
# To login to the website.
def show(email, password):
mycursor.execute('select email,password from userdata')
mylist = mycursor.fetchall()
for x in range(len(mylist)):
if mylist[x][0] == email and mylist[x][1] == password:
return True
# Check Duplicate.
def check_duplicate(email):
mycursor.execute('select email from userdata')
__author__ = 'Hossein Zolfi <*****@*****.**>'
import logging
from logging.handlers import RotatingFileHandler
from flask import Flask
from flask_sqlalchemy import SQLAlchemy
from flask.ext.script import Manager
from flask.ext.migrate import Migrate, MigrateCommand
from user_activation import *
from config import config_db
app = Flask(__name__, template_folder='../templates')
app.config.from_pyfile('../config/app.cfg')
config_db(app)
handler = RotatingFileHandler('app.log', maxBytes=100000, backupCount=1)
handler.setLevel(logging.INFO)
app.logger.addHandler(handler)
db = SQLAlchemy(app)
migrate = Migrate(app, db)
manager = Manager(app)
manager.add_command('db', MigrateCommand)
