def main():
# db_file_name = input("DB filename: ").strip() or ".temp.db"
# organizer_root = input("Gallery root: ").strip() or "data"
db_file_name = ".temp.db"
organizer_root = "data"
# os.remove(db_file_name)
organizer = Organizer(organizer_root, db_file_name)
organizer.analyze_root()
def test_organizer_entity_scanning(sample_root_path, tmp_path):
organizer = Organizer(sample_root_path, tmp_path / "sample.db")
organizer.analyze_root()
assert len(organizer.get_all_entities()) == 9
names = {(entity.name, entity.type_)
for entity in organizer.get_all_entities().values()}
assert names == {
("file1", EntityTypeId.Unkown),
("file2.txt", EntityTypeId.Unkown),
("file3.jpg", EntityTypeId.Image),
("file4.gif", EntityTypeId.Gif),
("file5.mp4", EntityTypeId.Video),
("file6.png", EntityTypeId.Image),
("sub1", EntityTypeId.Gallery),
("sub2file1", EntityTypeId.Unkown),
("sub2file2.flv", EntityTypeId.Video),
}
