Skip to content
GitLab
An error occurred while loading the file. Please try again.
Switch branch/tag
Find file BlameHistoryPermalink
reset_obs.py 3.42 KiB
Open in Web IDE
1 
from flask import Flask, render_template, request, redirect
2 
import yaml
3 
import ast
4
5 
app = Flask(__name__)
6
7
8 
key_type_map = {
9 
    'obs_type': str,
10 
    'obs_type_code': str,
11 
    'obs_id': str,
12 
    'run_chips': list,
13 
    'call_sequence': {
14 
        'scie_obs': {
15 
          'shutter_effect': bool,
16 
          'flat_fielding': bool,
17 
          'field_dist': bool,
18 
        },
19 
        'sky_background': {
20 
            'shutter_effect': bool,
21 
            'flat_fielding': bool,
22 
            'enable_straylight_model': bool,
23 
            'flat_level': None,
24 
            'flat_level_filt': None,
25 
        },
26 
        'PRNU_effect': {},
27 
        'cosmic_rays': {
28 
            'save_cosmic_img':  bool,
29 
        },
30 
        'poisson_and_dark': {
31 
            'add_dark': bool,
32 
        },
33 
        'bright_fatter': {},
34 
        'detector_defects': {
35 
            'hot_pixels': bool,
36 
            'dead_pixels': bool,
37 
            'bad_columns': bool,
38 
        },
39 
        'nonlinearity': {},
40 
        'blooming': {},
41 
        'prescan_overscan': {
42 
            'add_dark': bool,
43 
        },
44 
        'bias': {
45 
            'bias_16channel': bool,
46 
        },
47 
        'readout_noise': {},
48 
        'gain': {
49 
            'gain_16channel': bool,
50 
        },
51 
        'quantization_and_output': {
52 
            'format_output': bool,
53 
        },
54 
    },
55 
}
56
57
58 
def convert_dict_values(d, key_type_map):
59 
    for key, value in d.items():
60 
        if isinstance(value, dict):
61 
            convert_dict_values(value, key_type_map[key])
62 
        elif key in key_type_map:
63 
            if key_type_map[key] is int:
64 
                d[key] = int(value)
65 
            if key_type_map[key] is float:
66 
                d[key] = float(value)
67 
            if key_type_map[key] is bool:
68 
                if d[key].lower() == 'yes' or d[key].lower() == 'true':
69 
                    d[key] = True
70 
                else:
71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123
d[key] = False if key_type_map[key] is str: d[key] = str(value) if key_type_map[key] is list: d[key] = ast.literal_eval(value) if key_type_map[key] is None: d[key] = None def load_yaml(): with open('templates/obs_config_SCI.yaml', 'r') as file: return yaml.safe_load(file) def save_yaml(data): convert_dict_values(data, key_type_map) with open('config_reset/obs_config_SCI_reset.yaml', 'w') as file: yaml.dump(data, file, default_flow_style=False, sort_keys=False) def render_form(data, parent_key=''): form_html = '' for key, value in data.items(): full_key = f"{parent_key}.{key}" if parent_key else key if isinstance(value, dict): # 处理字典 form_html += f"<div class='block'><h2>{key}</h2>{render_form(value, full_key)}</div>" else: form_html += f"<label for='{full_key}'>{key}:</label>" form_html += f"<input type='text' id='{full_key}' name='{full_key}' value='{value}'><br>" return form_html @app.route('/', methods=['GET', 'POST']) def index(): if request.method == 'POST': data = load_yaml() for key in request.form: keys = key.split('.') temp = data for k in keys[:-1]: temp = temp[k] temp[keys[-1]] = request.form[key] save_yaml(data) return redirect('/') data = load_yaml() form_html = render_form(data) return render_template('index_obs.html', form_html=form_html) if __name__ == '__main__': app.run(debug=True)