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)