Initial commit.

This commit is contained in:
Martin Brodbeck 2024-07-26 13:22:44 +02:00
commit 89978aef79

93
fix_tolino_bool.py Normal file
View file

@ -0,0 +1,93 @@
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.")