Coverage for src/sqlqb/mysql/__init__.py: 0%

1from pymysql.connections import Connection as _Connection 

2from pymysql.cursors import DictCursor 

3 

4from sqlqb import Select as _Select 

5 

6 

7class Select(_Select): 

8 def __init__(self, connection, *args, **kwargs): 

9 self.__connection = connection 

10 super().__init__(*args, **kwargs) 

11 

12 def fetchone(self, cursor=None): 

13 cursor = cursor or self.__connection.cursor() 

14 with cursor as cursor: 

15 cursor.execute(self.sql, self.params) 

16 result = cursor.fetchone() 

17 return result 

18 

19 def fetchall(self, cursor=None): 

20 cursor = cursor or self.__connection.cursor() 

21 with cursor as cursor: 

22 cursor.execute(self.sql, self.params) 

23 result = cursor.fetchall() 

24 return result 

25 

26 

27class Connection(_Connection): 

28 def __init__(self, *args, **kwargs): 

29 if "cursorclass" not in kwargs: 

30 kwargs["cursorclass"] = DictCursor 

31 super().__init__(*args, **kwargs) 

32 

33 def Select(self, *args, **kwargs): 

34 return Select(self, *args, **kwargs)