From 2d80b69afeb43ad449e6ab1ec9e65f2f34f5c5dd Mon Sep 17 00:00:00 2001
From: Mathieu Loiseau <mathieu.loiseau@univ-grenoble-alpes.fr>
Date: Fri, 30 Apr 2021 11:14:18 +0200
Subject: [PATCH] DB connection
---
extract_notes.py | 20 ++++++++++++++++++--
1 file changed, 18 insertions(+), 2 deletions(-)
diff --git a/extract_notes.py b/extract_notes.py
index 6c9c726..759158d 100755
--- a/extract_notes.py
+++ b/extract_notes.py
@@ -2,8 +2,9 @@
# -*- coding: UTF-8 -*-
# created by lzbk
import argparse
-import os ,sys
+import os, sys, os.path
import getpass
+import sqlite3
parser = argparse.ArgumentParser(description="Extraire les notes d'une liseuse Vivlio")
parser.add_argument("-s", "--source", help="le dossier de la liseuse", type=str, default=None)
@@ -29,11 +30,26 @@ def getSource():
res=None
return res
+def connect(sourceDir):
+ try:
+ path = sourceDir+'/system/config/books.db'
+ if not os.path.isfile(path):
+ raise ValueError ("Failed to connect to DB ("+path+").")
+ else:
+ conn = sqlite3.connect(path)
+ print("Connected to '"+path+"'.")
+ except (ValueError,sqlite3.Error) as e:
+ print (str(e))
+ conn = None
+ return conn
#main
if __name__ == "__main__":
if args.source == None:
source = getSource()
- print(source)
else:
source = args.source
+ if source != None:
+ conn = connect(source)
+ if conn != None:
+ conn.close()
--
GitLab