[bde] Création de base.py pour factorisation de getcursor()

[scripts-20-100.git] / bde / base.py

diff --git a/bde/base.py b/bde/base.py
new file mode 100644 (file)
index 0000000..1a3fa65
--- /dev/null
+++ b/bde/base.py
@@ -0,0 +1,18 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+Des fonctions que j'utilise partout dans mes scripts bde/note.
+
+Parce que la duplication de code, c'est le mal.
+"""
+
+import psycopg2
+import psycopg2.extras
+
+def getcursor():
+    """Ouvre une connexion à la base de données de la note."""
+    con = psycopg2.connect(database="note")
+    con.set_client_encoding("utf-8")
+    cur = con.cursor(cursor_factory = psycopg2.extras.DictCursor)
+    return (con, cur)