ruff config

ASFHyP3 · Dec 17, 2024 · dfd0271 · dfd0271
1 parent 219ac15
commit dfd0271
Show file tree

Hide file tree

Showing 13 changed files with 1,226 additions and 1,457 deletions.
diff --git a/image_services/egis/create_configs.py b/image_services/egis/create_configs.py
@@ -2,61 +2,62 @@
 
 from jinja2 import Environment, PackageLoader, StrictUndefined, select_autoescape
 
+
 SEASONS = {
-    "JJA": {
-        "Season": "summer",
-        "SeasonAbbrev": "Jun/Jul/Aug",
-        "SeasonFull": "June/July/August",
+    'JJA': {
+        'Season': 'summer',
+        'SeasonAbbrev': 'Jun/Jul/Aug',
+        'SeasonFull': 'June/July/August',
     },
-    "SON": {
-        "Season": "fall",
-        "SeasonAbbrev": "Sep/Oct/Nov",
-        "SeasonFull": "September/October/November",
+    'SON': {
+        'Season': 'fall',
+        'SeasonAbbrev': 'Sep/Oct/Nov',
+        'SeasonFull': 'September/October/November',
     },
-    "DJF": {
-        "Season": "winter",
-        "SeasonAbbrev": "Dec/Jan/Feb",
-        "SeasonFull": "December/January/February",
+    'DJF': {
+        'Season': 'winter',
+        'SeasonAbbrev': 'Dec/Jan/Feb',
+        'SeasonFull': 'December/January/February',
     },
-    "MAM": {
-        "Season": "spring",
-        "SeasonAbbrev": "Mar/Apr/May",
-        "SeasonFull": "March/April/May",
+    'MAM': {
+        'Season': 'spring',
+        'SeasonAbbrev': 'Mar/Apr/May',
+        'SeasonFull': 'March/April/May',
     },
 }
 
 
 def make_configuration(data_type, polarization, season):
     config = {
-        "project_name": "GSSICB",
-        "s3_prefix": "tiles/",
-        "s3_suffix": f"_{SEASONS[season]['Season']}_{polarization.lower()}_{data_type}.tif",
-        "dataset_name": f"{data_type}_{polarization.upper()}_{season}",
-        "raster_function_templates": [
-            "ScaledCoherence.rft.xml",
-            "UnscaledCoherence.rft.xml",
+        'project_name': 'GSSICB',
+        's3_prefix': 'tiles/',
+        's3_suffix': f"_{SEASONS[season]['Season']}_{polarization.lower()}_{data_type}.tif",
+        'dataset_name': f'{data_type}_{polarization.upper()}_{season}',
+        'raster_function_templates': [
+            'ScaledCoherence.rft.xml',
+            'UnscaledCoherence.rft.xml',
         ],
-        "default_raster_function_template": "UnscaledCoherence.rft.xml",
+        'default_raster_function_template': 'UnscaledCoherence.rft.xml',
     }
     return config
 
 
 def make_metadata_fields(data_type, polarization, season):
     metadata = {
-        "data_type": data_type,
-        "interval": int(data_type[-2:]),
-        "polarization": polarization,
-        "months_abbreviated": SEASONS[season]["SeasonAbbrev"],
-        "season": SEASONS[season]["Season"],
-        "months_full": SEASONS[season]["SeasonFull"],
+        'data_type': data_type,
+        'interval': int(data_type[-2:]),
+        'polarization': polarization,
+        'months_abbreviated': SEASONS[season]['SeasonAbbrev'],
+        'season': SEASONS[season]['Season'],
+        'months_full': SEASONS[season]['SeasonFull'],
     }
     return metadata
 
 
 def get_environment() -> Environment:
     env = Environment(
-        loader=PackageLoader("metadata", "templates"),
-        autoescape=select_autoescape(["html.j2", "xml.j2"]),
+        loader=PackageLoader('metadata', 'templates'),
+        autoescape=select_autoescape(['html.j2', 'xml.j2']),
         undefined=StrictUndefined,
         trim_blocks=True,
         lstrip_blocks=True,
@@ -72,21 +73,19 @@ def render_template(template: str, payload: dict) -> str:
     return rendered
 
 
-data_types = ["COH24"]
-polarizations = ["VV", "HH"]
+data_types = ['COH24']
+polarizations = ['VV', 'HH']
 
 for data_type in data_types:
     for polarization in polarizations:
         for season in SEASONS:
-            service_name = f"{data_type}_{polarization.upper()}_{season}"
+            service_name = f'{data_type}_{polarization.upper()}_{season}'
             config_dict = make_configuration(data_type, polarization, season)
             metadata_dict = make_metadata_fields(data_type, polarization, season)
 
-            with open(f"config/{service_name}.json", "w") as f:
+            with open(f'config/{service_name}.json', 'w') as f:
                 json.dump(config_dict, f, indent=2)
 
-            output_metadata = render_template("coherence.txt.j2", metadata_dict)
-            with open(
-                f"metadata/service_metadata/{service_name}_metadata.txt", "w"
-            ) as f:
+            output_metadata = render_template('coherence.txt.j2', metadata_dict)
+            with open(f'metadata/service_metadata/{service_name}_metadata.txt', 'w') as f:
                 f.write(output_metadata)