Skip to content

Insert Data Script

Utility script for inserting demo person data into the database.

Functions

app.insertdata._build_demo_person 

_build_demo_person()

Build a demo Person instance with predefined values.

Creates a Person object with hardcoded demo data for testing purposes.

Returns:

Name Type Description
Person Person

A Person instance with demo data (name: "Demo", surname: "User", birthday: 2000-01-01, role: GUEST).
Source code in backend/app/insertdata.py 
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
def _build_demo_person() -> Person:
    """Build a demo Person instance with predefined values.

    Creates a Person object with hardcoded demo data for testing purposes.

    Returns:
        Person: A Person instance with demo data (name: "Demo", 
            surname: "User", birthday: 2000-01-01, role: GUEST).

    """
    return Person(
        name="Demo",
        surname="User",
        birthday=datetime(2000, 1, 1),
        role=RoleType.GUEST,
    )

app.insertdata.main 

main()

Main function to insert demo person data into the database.

Analyzes all images in the configured images folder, extracts face encodings, and inserts a demo person with the collected encodings into the database. Logs errors if no valid encodings are found or if database insertion fails.

Source code in backend/app/insertdata.py 
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
def main() -> None:
    """Main function to insert demo person data into the database.

    Analyzes all images in the configured images folder, extracts face encodings,
    and inserts a demo person with the collected encodings into the database.
    Logs errors if no valid encodings are found or if database insertion fails.

    """
    dataset = database.Database(
        url=set.url,
        name=set.name,
        collection=set.collection,
    )
    people = dataset.get_all_people() 
    engine = FaceEngine(people)   
    demo_person = _build_demo_person()

    all_encodings = {} 
    folder = Path(path_settings.imgsfolder)

    for elemento in folder.iterdir():
        if elemento.is_file() and not elemento.name.startswith('.'):
            print(f"Analisi di: {elemento.name}...")
            new_data = engine.analyze_img(elemento)

            if new_data is not None:
                all_encodings.update(new_data) 
                print(f" -> OK, volto trovato.")
            else:
                print(f" -> Nessun volto trovato o file non valido.")

    if not all_encodings:
        logger.error("Nessun encoding valido generato da nessuna foto.")
        return

    demo_person.encoding = all_encodings
    saved = dataset.add_person(demo_person)

    if saved is not None:
        logger.info(f"Inserimento completato. ID: {saved.id} - Totale foto: {len(all_encodings)}")
        print("Inserimento completato.")
    else:
        logger.error("Inserimento fallito da parte del Database.")
        print("Inserimento fallito.")