93 lines
3.5 KiB
Python
93 lines
3.5 KiB
Python
import argparse
|
|
import sqlite3
|
|
from os.path import exists, dirname, join, isfile
|
|
import sys
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("mount_point", help="Directory/Mount point of your Tolino device", type=str)
|
|
parser.add_argument("-n", "--dry-run", help="Don't change database; show information only.", action="store_true")
|
|
args = parser.parse_args()
|
|
|
|
if not exists(args.mount_point):
|
|
print("Directory/Mount point does not exist. Exiting.")
|
|
sys.exit(1)
|
|
|
|
dir_path = dirname(args.mount_point)
|
|
version_file = join(dir_path, ".kobo", "version")
|
|
device_version_info = ""
|
|
if isfile(version_file):
|
|
with open(version_file) as vf:
|
|
device_version_info = vf.read().strip().split(",")
|
|
else:
|
|
print("No device information found. Please sprecify the database file directly on the device.\nExiting.")
|
|
sys.exit(1)
|
|
device_model_id = device_version_info[-1]
|
|
if device_model_id.endswith("691"):
|
|
print("Tolino Shine 5 found.")
|
|
elif device_model_id.endswith("693"):
|
|
print("Tolino Shine Color found.")
|
|
elif device_model_id.endswith("690"):
|
|
print("Tolino Vision Color found.")
|
|
else:
|
|
print("No supported Tolino device found. Exiting.")
|
|
sys.exit(1)
|
|
|
|
dbfile = join(dir_path, ".kobo", "KoboReader.sqlite")
|
|
con = sqlite3.connect(dbfile)
|
|
cur = con.cursor()
|
|
|
|
res = cur.execute("SELECT version FROM DbVersion")
|
|
item = res.fetchone()
|
|
if item is None:
|
|
print("No valid database file. Exiting.")
|
|
sys.exit(1)
|
|
version = item[0]
|
|
if version < 188:
|
|
print("Database version unsupported. Exiting.")
|
|
sys.exit(1)
|
|
|
|
res = cur.execute("SELECT Id FROM Shelf WHERE (_IsDeleted <> true AND _IsDeleted <> false) OR (_IsVisible <> true AND _IsVisible <> false) OR (_IsSynced <> true AND _IsSynced <> false)")
|
|
items_shelf = res.fetchall()
|
|
|
|
res = cur.execute("SELECT ShelfName, ContentId FROM ShelfContent WHERE (_IsDeleted <> true AND _IsDeleted <> false) OR (_IsSynced <> true AND _IsSynced <> false)")
|
|
items_shelf_content = res.fetchall()
|
|
|
|
|
|
if len(items_shelf) == 0:
|
|
print("No affected entries in table Shelf.")
|
|
else:
|
|
print("Affected entries in table Shelf:")
|
|
for item in items_shelf:
|
|
print(f"- {item[0]}")
|
|
|
|
if len(items_shelf_content) == 0:
|
|
print("No affected entries in table ShelfContent.")
|
|
else:
|
|
print("Affected entries in table Shelf:")
|
|
for item in items_shelf_content:
|
|
print(f"- {item[0]} with ContentId {item[1]}")
|
|
|
|
if len(items_shelf) == 0 and len(items_shelf_content) == 0:
|
|
print("Database is ok.")
|
|
sys.exit(0)
|
|
|
|
if len(items_shelf) > 0 and not args.dry_run:
|
|
print("Fixing table 'Shelf' ...")
|
|
cur.execute("UPDATE Shelf SET _IsDeleted = true WHERE _IsDeleted = 'true'")
|
|
cur.execute("UPDATE Shelf SET _IsVisible = true WHERE _IsVisible = 'true'")
|
|
cur.execute("UPDATE Shelf SET _IsSynced = true WHERE _IsSynced = 'true'")
|
|
cur.execute("UPDATE Shelf SET _IsDeleted = false WHERE _IsDeleted = 'false'")
|
|
cur.execute("UPDATE Shelf SET _IsVisible = false WHERE _IsVisible = 'false'")
|
|
cur.execute("UPDATE Shelf SET _IsSynced = false WHERE _IsSynced = 'false'")
|
|
con.commit()
|
|
|
|
if len(items_shelf_content) > 0 and not args.dry_run:
|
|
print("Fixing table 'ShelfContent' ...")
|
|
cur.execute("UPDATE ShelfContent SET _IsDeleted = true WHERE _IsDeleted = 'true'")
|
|
cur.execute("UPDATE ShelfContent SET _IsSynced = true WHERE _IsSynced = 'true'")
|
|
cur.execute("UPDATE ShelfContent SET _IsDeleted = false WHERE _IsDeleted = 'false'")
|
|
cur.execute("UPDATE ShelfContent SET _IsSynced = false WHERE _IsSynced = 'false'")
|
|
con.commit()
|
|
|
|
if not args.dry_run:
|
|
print("Done.")
|