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), }