18 lines
494 B
Python
18 lines
494 B
Python
|
# vim: set fileencoding=utf-8
|
||
|
|
from typing import List, Dict, Any
|
||
|
|
|
||
|
|
|
||
|
|
class FakeCursor():
|
||
|
|
|
||
|
|
def __init__(self, rows: List[Dict[str, Any]]) -> None:
|
||
|
|
self.__rows = rows
|
||
|
|
self.rowcount = len(rows)
|
||
|
|
|
||
|
|
def fetchone(self) -> Dict[str, Any]:
|
||
|
|
if len(self.__rows) != 1:
|
||
|
|
raise Exception('Tried to fetch one row and there are {} rows!'.format(len(self.__rows)))
|
||
|
|
return self.__rows[0]
|
||
|
|
|
||
|
|
def fetchall(self) -> List[Dict[str, Any]]:
|
||
|
|
return self.__rows
|