Skip to content

Commit e8357e9

Browse files
committed
new insert query
1 parent 78521a7 commit e8357e9

File tree

2 files changed

+21
-3
lines changed

2 files changed

+21
-3
lines changed

sql.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,24 @@ def insertQuery(query):
3737
database=db["DB"]
3838
)
3939

40+
mycursor = mydb.cursor()
41+
mycursor.execute(query)
42+
43+
mydb.commit()
44+
45+
return mycursor.rowcount
46+
47+
def insert_escaped_query(query):
48+
"""Executes `INSERT` query `mycursor.execute("", (query))` provided and returns `mycursor.rowcount`\n
49+
Connects to a predefined `Database` from `config.json`"""
50+
db = config["DATABASE"]
51+
mydb = mysql.connector.connect(
52+
host=db["HOST"],
53+
user=db["USERNAME"],
54+
password=db["PASSWORD"],
55+
database=db["DB"]
56+
)
57+
4058
mycursor = mydb.cursor()
4159
mycursor.execute("", (query))
4260

surftimer/ck_playerrank.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from fastapi import APIRouter, Request, Response, status
22
from fastapi.responses import JSONResponse
3-
from sql import selectQuery, insertQuery
3+
from sql import selectQuery, insertQuery, insert_escaped_query
44
from globals import get_cache, set_cache
55
from pydantic import BaseModel
66
import time, json
@@ -128,7 +128,7 @@ async def updatePlayerRankPoints(
128128
data.style,
129129
)
130130

131-
xquery = insertQuery(sql)
131+
xquery = insert_escaped_query(sql)
132132

133133
content_data = {"updated": xquery, "xtime": time.perf_counter() - tic}
134134
if xquery < 1:
@@ -184,7 +184,7 @@ async def updatePlayerRankPoints2(
184184
data.steamid32,
185185
data.style,
186186
)
187-
xquery = insertQuery(sql)
187+
xquery = insert_escaped_query(sql)
188188

189189
content_data = {"updated": xquery, "xtime": time.perf_counter() - tic}
190190
if xquery < 1:

0 commit comments

Comments
 (0)