rework main models

dbapp/mapsapp/utils.py

@@ -3,6 +3,7 @@ import re
 import json
 from .models import Transponders
 from mainapp.models import Polarization, Satellite
+from io import BytesIO
 
 def search_satellite_on_page(data: dict, satellite_name: str):
     for pos, value in data.get('page', {}).get('positions').items():
@@ -90,3 +91,68 @@ def parse_transponders_from_json(filepath: str):
                 )
                 tran_obj.save()
                 
+
+from lxml import etree
+
+def parse_transponders_from_xml(data_in: BytesIO):
+    
+    tree = etree.parse(data_in)
+    ns = {
+        'i': 'http://www.w3.org/2001/XMLSchema-instance', 
+        'ns': 'http://schemas.datacontract.org/2004/07/Geolocation.Domain.Utils.Repository.SatellitesSerialization.Memos',
+        'tr': 'http://schemas.datacontract.org/2004/07/Geolocation.Common.Extensions'
+    }
+    satellites = tree.xpath('//ns:SatelliteMemo', namespaces=ns)
+    for sat in satellites[:]:
+        name = sat.xpath('./ns:name/text()', namespaces=ns)[0]
+        if name == 'X' or 'DONT USE' in name:
+            continue
+        norad = sat.xpath('./ns:norad/text()', namespaces=ns)
+        beams = sat.xpath('.//ns:BeamMemo', namespaces=ns)
+        zones = {}
+        for zone in beams:
+            zone_name = zone.xpath('./ns:name/text()', namespaces=ns)[0] if zone.xpath('./ns:name/text()', namespaces=ns) else '-'
+            zones[zone.xpath('./ns:id/text()', namespaces=ns)[0]] = {
+                "name": zone_name,
+                "pol": zone.xpath('./ns:polarization/text()', namespaces=ns)[0],
+            }
+        transponders = sat.xpath('.//ns:TransponderMemo', namespaces=ns)
+        for transponder in transponders:
+            tr_id = transponder.xpath('./ns:downlinkBeamId/text()', namespaces=ns)[0]
+            downlink_start = float(transponder.xpath('./ns:downlinkFrequency/tr:start/text()', namespaces=ns)[0])
+            downlink_end = float(transponder.xpath('./ns:downlinkFrequency/tr:end/text()', namespaces=ns)[0])
+            uplink_start = float(transponder.xpath('./ns:uplinkFrequency/tr:start/text()', namespaces=ns)[0])
+            uplink_end = float(transponder.xpath('./ns:uplinkFrequency/tr:end/text()', namespaces=ns)[0])
+            tr_data = zones[tr_id]
+            # p = tr_data['pol'][0] if tr_data['pol'] else '-'
+            match tr_data['pol']:
+                case 'Horizontal':
+                    pol = 'Горизонтальная'
+                case 'Vertical':
+                    pol = 'Вертикальная'
+                case 'CircularRight':
+                    pol = 'Правая'
+                case 'CircularLeft':
+                    pol = 'Левая'
+                case _:
+                    pol = '-'
+            tr_name = transponder.xpath('./ns:name/text()', namespaces=ns)[0]
+
+            pol_obj, _ = Polarization.objects.get_or_create(name=pol)
+            sat_obj, _ = Satellite.objects.get_or_create(
+                name=name,
+                defaults={
+                    "norad": int(norad[0]) if norad else -1
+            })
+            trans_obj, _ = Transponders.objects.get_or_create(
+                polarization=pol_obj,
+                downlink=(downlink_start+downlink_end)/2/1000000,
+                uplink=(uplink_start+uplink_end)/2/1000000,
+                frequency_range=abs(downlink_end-downlink_start)/1000000,
+                name=tr_name,
+                defaults={
+                    "zone_name": tr_data['name'],
+                    "sat_id": sat_obj,
+                }
+            )
+            trans_obj.save()