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)