def db(self, db: dict):
"""Set the Database parameters from a dict."""
db.setdefault("version", 1) # legacy dbs are version 1
db.setdefault("hash_algorithm", "sha256") # legacy dbs use sha256
db.setdefault("timezone_default", "local") # legacy dbs are in local time
db["version"] = int(db["version"])
if db["version"] > self.VERSION:
raise DatabaseException(
"Database version too new for this version of PhotoManager."
)
if db["version"] < 3:
for uid in db["photo_db"].keys():
photos = db["photo_db"][uid]
for i in range(len(photos)):
photos[i] = {NAME_MAP_ENC[k]: v for k, v in photos[i].items()}
db = {k: db[k] for k in self.DB_KEY_ORDER}
db["hash_algorithm"] = HashAlgorithm(db["hash_algorithm"])
for uid in db["photo_db"].keys():
db["photo_db"][uid] = [PhotoFile.from_dict(d) for d in db["photo_db"][uid]]
db["version"] = self.VERSION
self._db = db
for uid, photos in self.photo_db.items():
for photo in photos:
self.hash_to_uid[photo.chk] = uid
if photo.ts in self.timestamp_to_uids:
self.timestamp_to_uids[photo.ts][uid] = None
else:
self.timestamp_to_uids[photo.ts] = {uid: None}
self.reset_saved()
def test_photofile_from_file(datafiles):
with ExifTool():
for pf in photofile_expected_results:
pf = PhotoFile.from_dict(pf.to_dict())
rel_path = pf.src
pf.src = str(datafiles / rel_path)
new_pf = PhotoFile.from_file(
pf.src,
tz_default=timezone(timedelta(seconds=pf.tzo)),
)
assert new_pf == pf
def test_database_load_version_1():
json_data = b"""{
"version": 1,
"hash_algorithm": "sha256",
"photo_db": {
"d239210f00534b76a2b215e073f75832": [
{
"checksum": "deadbeef",
"source_path": "/a/b/c.jpg",
"datetime": "2015:08:27 04:09:36.50",
"timestamp": 1440662976.5,
"file_size": 1024,
"store_path": "/d/e/f.jpg",
"priority": 11
},
{
"checksum": "deadbeef",
"source_path": "/g/b/c.jpg",
"datetime": "2015:08:27 04:09:36.50",
"timestamp": 1440662976.5,
"file_size": 1024,
"store_path": "",
"priority": 20,
"tz_offset": -14400
}
]
},
"command_history": {
"2021-03-08_23-56-00Z": "photomanager create --db test.json",
"2021-03-08_23-57-00Z": "photomanager import --db test.json test.jpg"
}
}"""
db = Database.from_json(json_data)
print(db.db)
assert db.version == Database.VERSION
assert db.hash_algorithm == HashAlgorithm.SHA256
assert db.db["timezone_default"] == "local"
assert db.timezone_default is None
photo_db_expected = {
"d239210f00534b76a2b215e073f75832": [
PhotoFile.from_dict({
"chk": "deadbeef",
"src": "/a/b/c.jpg",
"dt": "2015:08:27 04:09:36.50",
"ts": 1440662976.5,
"fsz": 1024,
"sto": "/d/e/f.jpg",
"prio": 11,
}),
PhotoFile.from_dict({
"chk": "deadbeef",
"src": "/g/b/c.jpg",
"dt": "2015:08:27 04:09:36.50",
"ts": 1440662976.5,
"fsz": 1024,
"sto": "",
"prio": 20,
"tzo": -14400,
}),
]
}
command_history_expected = {
"2021-03-08_23-56-00Z": "photomanager create --db test.json",
"2021-03-08_23-57-00Z": "photomanager import --db test.json test.jpg",
}
db_expected = {
"version": Database.VERSION,
"hash_algorithm": HashAlgorithm.SHA256,
"timezone_default": "local",
"photo_db": photo_db_expected,
"command_history": command_history_expected,
}
assert db.photo_db == photo_db_expected
assert db.command_history == command_history_expected
assert orjson.loads(db.json) != orjson.loads(json_data)
assert db.db == db_expected
assert db == Database.from_dict(orjson.loads(json_data))
assert db.get_stats() == (1, 2, 1, 1024)
def test_database_load_version_3():
json_data = b"""{
"version": 3,
"hash_algorithm": "sha256",
"photo_db": {
"QKEsTn2X": [
{
"chk": "deadbeef",
"src": "/a/b/c.jpg",
"dt": "2015:08:27 04:09:36.50",
"ts": 1440662976.5,
"fsz": 1024,
"sto": "/d/e/f.jpg",
"prio": 11,
"tzo": null
},
{
"chk": "deadbeef",
"src": "/g/b/c.jpg",
"dt": "2015:08:27 04:09:36.50",
"ts": 1440662976.5,
"fsz": 1024,
"sto": "",
"prio": 20,
"tzo": -14400
}
]
},
"command_history": {
"2021-03-08_23-56-00Z": "photomanager create --db test.json",
"2021-03-08_23-57-00Z": "photomanager import --db test.json test.jpg"
}
}""".replace(b"VERSION", f"{Database.VERSION}".encode())
db = Database.from_json(json_data)
print(db.db)
assert db.version == Database.VERSION
assert db.hash_algorithm == HashAlgorithm.SHA256
assert db.db["timezone_default"] == "local"
assert db.timezone_default is None
photo_db_expected = {
"QKEsTn2X": [
PhotoFile.from_dict({
"chk": "deadbeef",
"src": "/a/b/c.jpg",
"dt": "2015:08:27 04:09:36.50",
"ts": 1440662976.5,
"fsz": 1024,
"sto": "/d/e/f.jpg",
"prio": 11,
}),
PhotoFile.from_dict({
"chk": "deadbeef",
"src": "/g/b/c.jpg",
"dt": "2015:08:27 04:09:36.50",
"ts": 1440662976.5,
"fsz": 1024,
"sto": "",
"prio": 20,
"tzo": -14400,
}),
]
}
command_history_expected = {
"2021-03-08_23-56-00Z": "photomanager create --db test.json",
"2021-03-08_23-57-00Z": "photomanager import --db test.json test.jpg",
}
db_expected = {
"version": Database.VERSION,
"hash_algorithm": HashAlgorithm.SHA256,
"timezone_default": "local",
"photo_db": photo_db_expected,
"command_history": command_history_expected,
}
assert db.photo_db == photo_db_expected
assert db.command_history == command_history_expected
assert orjson.loads(db.json) != orjson.loads(json_data)
assert db.db == db_expected
assert db == Database.from_dict(orjson.loads(json_data))
assert db.get_stats() == (1, 2, 1, 1024)