-
Notifications
You must be signed in to change notification settings - Fork 0
/
interact_with_DB.py
41 lines (33 loc) · 1.4 KB
/
interact_with_DB.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
# in terminal: pip install mysql-connector-python
import mysql.connector
# ------------------------------------------------- #
# ------------- DATABASE CONNECTION --------------- #
# ------------------------------------------------- #
def interact_db(query, query_type: str):
return_value = False
connection = mysql.connector.connect(host='34.123.35.73',
user='root',
database='Hakolboynik')
cursor = connection.cursor(named_tuple=True)
cursor.execute(query)
#
if query_type == 'commit':
# Use for INSERT, UPDATE, DELETE statements.
# Returns: The number of rows affected by the query (a non-negative int).
connection.commit()
return_value = True
if query_type == 'fetch':
# Use for SELECT statement.
# Returns: False if the query failed, or the result of the query if it succeeded.
query_result = cursor.fetchall()
return_value = query_result
if query_type == 'fetchone':
# Use for SELECT statement.
# Returns: False if the query failed, or the result of the query if it succeeded.
query_result = cursor.fetchone()
return_value = query_result
connection.close()
cursor.close()
return return_value
# ------------------------------------------------- #
# ------------------------------------------------- #