def add_user(self, email, password):
salt = ph.get_salt()
hash = ph.get_hash(password + salt)
try:
self.db.users.insert({"email": email, "salt": salt, "hash": hash})
return True
except pymongo.errors.DuplicateKeyError:
return False
def add_table(self, table_name, owner_id):
_id = passwordhelper.get_hash(owner_id + table_name)[:5]
MOCK_TABLES.append({
"_id": _id,
"name": table_name,
"owner": owner_id,
"url": config.base_url + "newrequest/" + _id
})
return _id
def add_user(self, email, password):
salt = passwordhelper.get_salt()
hash = passwordhelper.get_hash(password + salt)
with self as connection:
query = f"INSERT INTO users (email, salt, hash)" \
"Values (%s, %s, %s);"
with connection.cursor() as cursor:
cursor.execute(query, (email, salt, hash))
connection.commit()
return True
return False
def add_request(self, table_id, time):
for table in MOCK_TABLES:
if table["_id"] == table_id:
break
else:
return False
for i, req in enumerate(MOCK_REQUESTS):
if req["table_id"] == table_id:
del MOCK_REQUESTS[i]
break
_id = passwordhelper.get_hash(table_id)[:5]
MOCK_REQUESTS.append({
"_id": _id,
"table_id": table["_id"],
"table_name": table["name"],
"time": time
})
return True
import datetime
import config
import dbconfig
import passwordhelper
from user import User
from dbhelper import DBHelper
MOCK_USERS = {'*****@*****.**': '123456'}
MOCK_TABLES = [{
"_id": passwordhelper.get_hash("1"),
"name": "1",
"owner": "*****@*****.**",
"url": "mockurl"
}]
MOCK_REQUESTS = [{
"_id": "1",
"table_id": "1",
"table_name": "1",
"time": datetime.datetime.now()
}]
class MockLoginDB(DBHelper):
def __init__(self, db_name):
super().__init__(dbconfig.db_user, dbconfig.db_password, db_name,
'localhost')
def get_user(self, email):
with self as connection:
query = "SELECT salt, hash FROM users WHERE email=%s;"
with connection.cursor() as cursor: