diff --git a/extract_notes.py b/extract_notes.py index 6c9c726d14a58f80f1c715f4d3eeb5cf0f011713..759158d24051692b5b311ec850b7b5d9182ae20c 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()