notebookVB

knihovny/medicus_db.py

@@ -22,7 +22,7 @@ class MedicusDB:
         cols = [d[0].strip().lower() for d in self.cur.description]
         return [dict(zip(cols, row)) for row in self.cur.fetchall()]
 
-    def get_active_registered_patients(self):
+    def get_active_registered_patients(self, as_dict=False):
         sql = """
         SELECT
             kar.rodcis,
@@ -36,7 +36,9 @@ class MedicusDB:
           AND kar.rodcis IS NOT NULL
           AND kar.rodcis <> ''
         """
-        return self.query(sql)  # or self.query_dict(sql)
+        if as_dict:
+            return self.query_dict(sql)
+        return self.query(sql)
 
     def close(self):
         self.conn.close()