Added red and blue to the warehouse.xlsx options

Author: karel26

plugins/mission/README.md

@@ -73,6 +73,7 @@ On export, the Excel file will be named `warehouse_<airport>.xlsx`, where airpor
 If you want to upload the warehouse information to any other airport, all you need to do is change the ICAO in its name.
 
 ### Upload a Warehouse Inventory to all Airbases of a Coalition
+If you add "RED" or "BLUE" instead of an ICAO, you can upload the warehouse inventory to all airbases of this coalition.
 You can also _remove_ the ICAO completely and upload a file named `warehouse.xlsx`.
 You will then be asked to which coalition you want to upload the warehouse inventory.
 Afterward, all warehouses of this coalition will be filled with the respective inventory information.

plugins/mission/commands.py

@@ -2524,7 +2524,7 @@ async def handle_warehouse_uploads(self, message: discord.Message):
 
         att = message.attachments[0]
         filename = att.filename.lower()
-        match = re.match(r'^warehouse-([a-zA-Z]{4})\.xlsx?$', filename)
+        match = re.match(r'^warehouse_([^.]*)\.xlsx?$', filename)
         if not match:
             coalition = await utils.selection(
                 ctx,
@@ -2537,6 +2537,9 @@ async def handle_warehouse_uploads(self, message: discord.Message):
 
             data = await server.send_to_dcs_sync({"command": "getMissionSituation"})
             airports = data.get('coalitions', {}).get(coalition, {}).get('airbases')
+        elif match.group(1).upper() in ['RED', 'BLUE']:
+            data = await server.send_to_dcs_sync({"command": "getMissionSituation"})
+            airports = data.get('coalitions', {}).get(match.group(1).upper(), {}).get('airbases')
         else:
             icao = match.group(1).upper()
             airport = next((x for x in server.current_mission.airbases if x.get('code', x.get('type')) == icao), None)