diff --git a/sqli/dao/student.py b/sqli/dao/student.py index d41ef885..60b6ec90 100644 --- a/sqli/dao/student.py +++ b/sqli/dao/student.py @@ -2,7 +2,6 @@ from aiopg.connection import Connection - class Student(NamedTuple): id: int name: str @@ -27,21 +26,18 @@ async def get_many(conn: Connection, limit: Optional[int] = None, q = 'SELECT id, name FROM students' params = {} if limit is not None: - q += ' LIMIT + %(limit)s ' + q += ' LIMIT %s' params['limit'] = limit if offset is not None: - q += ' OFFSET + %(offset)s ' + q += ' OFFSET %s' params['offset'] = offset async with conn.cursor() as cur: - await cur.execute(q, params) + await cur.execute(q, (params.get('limit'), params.get('offset'))) results = await cur.fetchall() return [Student.from_raw(r) for r in results] @staticmethod async def create(conn: Connection, name: str): - q = ("INSERT INTO students (name) " - "VALUES ('%(name)s')" % {'name': name}) + q = "INSERT INTO students (name) VALUES (%s)" async with conn.cursor() as cur: - await cur.execute(q) - - + await cur.execute(q, (name,))