def get_users_by_phone_number(phones_number) -> list:
query = f"SELECT user_id FROM {SQLITE_DB_TABLE} WHERE "
for i in range(len(phones_number)):
query += f"phone = '{phones_number[i]}' OR "
if i == len(phones_number) - 1:
query += f"phone = '{phones_number[i]}'"
logging.info(f"Func /db/queries/select.get_users_by_phone_number with {query}")
response = SQLite(SQLITE_DB_NAME).select_query(query)
return response
def initDb(self, name, encoding):
"""Initializing SQLite database"""
self.dbName = name
self.db = SQLite(self.dbName, encoding)
self.priority = Priority(self.db)
self.task = Task(self.db)
# for testing purposes
if self.task.count() == 0:
self.db.execSql(
"insert into {} (name, priority, deadline) values(?, ?, ?)".
format(self.task._tableName),
("Low Test", 1, date.today() + timedelta(2)))
self.db.commit()
def test_sqlite():
with SQLite(DB_PATH) as my_conn:
assert os.path.isfile(DB_PATH)
my_conn.execute("SELECT 1")
res = list(my_conn.fetchall()[0])
assert res == [1]
def insert_new_user(user_id, first_name, last_name, phone):
query = f"""INSERT OR IGNORE INTO {SQLITE_DB_TABLE}
(user_id, first_name, second_name, phone)
VALUES ({user_id}, '{first_name}', '{last_name}', '{phone}');"""
logging.info(f"Func /db/queries/insert.insert_new_user with {query}")
SQLite(SQLITE_DB_NAME).insert_query(query)
def setUp(self):
with threading.Lock():
self.dbName = "test.sqlite3"
self.db = SQLite(self.dbName)
self.createTable()
def get_user_by_id(user_id) -> int:
query = f"SELECT user_id FROM {SQLITE_DB_TABLE} WHERE user_id = {user_id};"
logging.info(f"Func /db/queries/select.get_user_by_id with {query}")
response = SQLite(SQLITE_DB_NAME).select_query(query)
if len(response) > 0:
return response[0][0]
