def storeRS_ID(self, rs_id):
db = db_handler.get_db(self.db_path)
try:
db_handler.init_db(db)
except OperationalError:
pass
rs_id_status = False
db.execute("INSERT INTO rs_id_tbl (rs_id, used) \
VALUES (?, ?)", [rs_id, rs_id_status])
db.commit()
db.close()
def add_surrogate_id_to_code(self, code, surrogate_id):
db = db_handler.get_db(self.db_path)
try:
db_handler.init_db(db)
except OperationalError:
pass
for code in self.query_db("select * from codes where code = ?;", [code]):
code_from_db = code["code"]
code_is_valid_and_unused = "!" in code_from_db
if (code_is_valid_and_unused):
db.execute("UPDATE codes SET code=? WHERE ID=? ;", [surrogate_id, code])
db.commit()
else:
raise Exception("Invalid code")
def storeSurrogateJSON(DictionaryToStore):
db = db_handler.get_db()
try:
db_handler.init_db(db)
except OperationalError:
pass
debug_log.info(DictionaryToStore)
for key in DictionaryToStore:
debug_log.info(key)
db.execute(
"INSERT INTO surrogate_and_user_mapping (user_id, surrogate_id) \
VALUES (?, ?)", [key, dumps(DictionaryToStore[key])])
db.commit()
def storeCode(self, code):
db = db_handler.get_db(self.db_path)
try:
db_handler.init_db(db)
except OperationalError:
pass
code_key = list(code.keys())[0]
code_value = code[code_key]
db.execute("INSERT INTO codes (ID,code) \
VALUES (?, ?)", [code_key, code_value])
db.commit()
debug_log.info("{} {}".format(code_key, code_value))
for code in self.query_db("select * from codes where ID = ?;", [code_key]):
debug_log.info(code["code"])
db.close()
def storeJSON(self, DictionaryToStore):
db = db_handler.get_db(self.db_path)
try:
db_handler.init_db(db)
except OperationalError:
pass
debug_log.info(DictionaryToStore)
for key in DictionaryToStore:
debug_log.info(key)
try:
db.execute("INSERT INTO storage (surrogate_id,json) \
VALUES (?, ?)", [key, dumps(DictionaryToStore[key])])
db.commit()
except IntegrityError as e:
db.execute("UPDATE storage SET json=? WHERE surrogate_id=? ;", [dumps(DictionaryToStore[key]), key])
db.commit()
def storeCodeUser(DictionaryToStore):
# {"code": "user_id"}
db = db_handler.get_db()
try:
db_handler.init_db(db)
except OperationalError:
pass
debug_log.info(DictionaryToStore)
for key in DictionaryToStore:
debug_log.info(key)
db.execute(
"INSERT INTO code_and_user_mapping (code, user_id) \
VALUES (?, ?)", [key, dumps(DictionaryToStore[key])])
db.commit()
def change_rs_id_status(self, rs_id, status):
db = db_handler.get_db(self.db_path)
try:
db_handler.init_db(db)
except OperationalError:
pass
for rs_id_object in self.query_db("select * from rs_id_tbl where rs_id = ?;", [rs_id]):
rs_id_from_db = rs_id_object["rs_id"]
status_from_db = bool(rs_id_object["used"])
status_is_unused = status_from_db == False
if (status_is_unused):
db.execute("UPDATE rs_id_tbl SET used=? WHERE rs_id=? ;", [status, rs_id])
db.commit()
db.close()
return True
else:
db.close()
return False
def store_session(self, DictionaryToStore):
db = db_handler.get_db(self.db_path)
try:
db_handler.init_db(db)
except OperationalError:
pass
debug_log.info(DictionaryToStore)
for key in DictionaryToStore:
debug_log.info(key)
try:
db.execute("INSERT INTO session_store (code,json) \
VALUES (?, ?)", [key, dumps(DictionaryToStore[key])])
db.commit()
db.close()
except IntegrityError as e:
db.execute("UPDATE session_store SET json=? WHERE code=? ;", [dumps(DictionaryToStore[key]), key])
db.commit()
db.close()
def storeJSON(DictionaryToStore):
db = db_handler.get_db()
try:
db_handler.init_db(db)
except OperationalError:
pass
debug_log.info(DictionaryToStore)
for key in DictionaryToStore:
debug_log.info(key)
# codes = {"jsons": {}}
# codes = {"jsons": {}}
try:
db.execute(
"INSERT INTO storage (ID,json) \
VALUES (?, ?)", [key, dumps(DictionaryToStore[key])])
db.commit()
except IntegrityError as e:
db.execute("UPDATE storage SET json=? WHERE ID=? ;",
[dumps(DictionaryToStore[key]), key])
db.commit()
def storeCSR_JSON(self, DictionaryToStore):
cr_id = DictionaryToStore["cr_id"]
rs_id = DictionaryToStore["rs_id"]
surrogate_id = DictionaryToStore["surrogate_id"]
slr_id = DictionaryToStore["slr_id"]
json = DictionaryToStore["json"]
db = db_handler.get_db(self.db_path)
try:
db_handler.init_db(db)
except OperationalError:
pass
debug_log.info(DictionaryToStore)
# debug_log.info(key)
try:
db.execute("INSERT INTO csr_storage (cr_id, surrogate_id, slr_id, rs_id, json) \
VALUES (?, ?, ?, ?, ?)", [cr_id, surrogate_id, slr_id, rs_id, dumps(json)])
db.commit()
except IntegrityError as e:
# db.execute("UPDATE csr_storage SET json=? WHERE cr_id=? ;", [dumps(DictionaryToStore[key]), key])
# db.commit()
db.rollback()
raise DetailedHTTPException(detail={"msg": "Adding CSR to the database has failed.",},
title="Failure in CSR storage", exception=e)
def init():
db.init_db()
return jsonify(200)
