native-fisher-py/populate_submodules.py at main · z3rone-org/native-fisher-py · GitHub

1
2
3
4
5
6
7
8
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
import os

base_path = "native_fisher_py/python/native_fisher_py/data"

filter_enums = [
    "activation_type", "mass_analyzer_type", "ms_order_type", "ionization_mode_type",
    "scan_mode_type", "polarity_type", "detector_type", "scan_data_type",
    "sector_scan_type", "field_free_region_type", "tri_state", "energy_type",
    "event_accurate_mass", "source_fragmentation_value_type", "compensation_voltage_type"
]

mapping = {
    "mass_analyzer_type": "MassAnalyzer",
    "ms_order_type": "MSOrder"
}

for name in filter_enums:
    cls_name = mapping.get(name, "".join([x.capitalize() for x in name.split("_")]))
    dir_path = os.path.join(base_path, "filter_enums")
    os.makedirs(dir_path, exist_ok=True)
    # Also create the leaf modules in `data/` directly as redirects
    with open(os.path.join(base_path, f"{name}.py"), "w") as f_leaf:
        f_leaf.write(f"import enum\nfrom .classes import {cls_name}\n{name} = {cls_name}\n{cls_name} = {cls_name}\nenum = enum\n")

    with open(os.path.join(dir_path, f"{name}.py"), "w") as f:
        f.write(f"import enum\nfrom ..classes import {cls_name}\n{name} = {cls_name}\n{cls_name} = {cls_name}\nenum = enum\n")

business_objects = [
    "instrument_data", "sample_information", "file_header", "file_error",
    "scan_event", "scan_events", "scan_statistics", "segmented_scan",
    "centroid_stream", "scan_dependents", "log_entry", "header_item",
    "status_log_values", "tune_data_values", "reaction", "scan",
    "chromatogram_signal", "mass_options", "range", "trace_type", "error_log_entry", "peak_options", "ft_average_options", "tolerance_units", "common_core_data_object", "auto_sampler_information", "file_type", "filter_accurate_mass", "raw_file_classification", "scan_dependent_details", "scan_filter", "sequence_file_writer", "sequence_info", "source_fragmentation_info_valid_type", "tray_shape"
]

for name in business_objects:
    cls_name = "".join([x.capitalize() for x in name.split("_")])
    if name == "mass_options": cls_name = "MassOptions"
    if name == "range": cls_name = "Range"
    if name == "trace_type": cls_name = "TraceType"
    if name == "tolerance_units": cls_name = "ToleranceUnits"
    if name == "common_core_data_object": cls_name = "CommonCoreDataObject"
    if name == "auto_sampler_information": cls_name = "AutoSamplerInformation"
    if name == "file_type": cls_name = "FileType"
    if name == "filter_accurate_mass": cls_name = "FilterAccurateMass"
    if name == "raw_file_classification": cls_name = "RawFileClassification"
    if name == "scan_dependent_details": cls_name = "ScanDependentDetails"
    if name == "scan_filter": cls_name = "ScanFilter"
    if name == "sequence_file_writer": cls_name = "SequenceFileWriter"
    if name == "sequence_info": cls_name = "SequenceInfo"
    if name == "source_fragmentation_info_valid_type": cls_name = "SourceFragmentationInfoValidType"
    if name == "tray_shape": cls_name = "TrayShape"

    dir_path = os.path.join(base_path, "business")
    os.makedirs(dir_path, exist_ok=True)

    # Also create the leaf modules in `data/` directly as redirects
    with open(os.path.join(base_path, f"{name}.py"), "w") as f_leaf:
        f_leaf.write(f"import enum\nfrom .classes import {cls_name}\n{name} = {cls_name}\n{cls_name} = {cls_name}\nenum = enum\n")

    with open(os.path.join(dir_path, f"{name}.py"), "w") as f:
        f.write(f"import enum\nfrom ..classes import {cls_name}\n{name} = {cls_name}\n{cls_name} = {cls_name}\nenum = enum\n")