Skip to content

nwb.py

subject_to_nwb(session_key)

Generate a dictionary object containing subject information.

Parameters:

Name Type Description Default
session_key dict

Key specifying one entry in element_animal.subject.Subject

required

Returns:

Type Description

pynwb.file.Subject: NWB object

Source code in element_animal/export/nwb.py
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
def subject_to_nwb(session_key: dict):
    """Generate a dictionary object containing subject information.

    Args:
        session_key (dict): Key specifying one entry in element_animal.subject.Subject

    Returns:
        pynwb.file.Subject: NWB object
    """
    subject_query = subject.Subject & session_key
    subject_query = subject_query.join(subject.Subject.Line, left=True)
    subject_query = subject_query.join(subject.Subject.Strain, left=True)
    subject_query = subject_query.join(subject.Subject.Source, left=True)
    subject_info = subject_query.fetch1()

    return pynwb.file.Subject(
        subject_id=subject_info["subject"],
        sex=subject_info["sex"],
        date_of_birth=datetime.combine(
            subject_info["subject_birth_date"],
            datetime.strptime("00:00:00", "%H:%M:%S").time(),
        ),
        description=json.dumps(subject_info, default=str),
        species=str((subject.Line & subject_query).fetch("species")),
        genotype=" x ".join(
            (subject.Line.Allele * subject.Subject.Line & subject_query).fetch("allele")
        ),
    )