azure_cdn: use subgroups, attempt 2

app/terraform/proxy/__init__.py

@@ -1,4 +1,5 @@
 import datetime
+from collections import defaultdict
 
 from app import app
 from app.extensions import db
@@ -8,6 +9,20 @@ from app.terraform import BaseAutomation
 
 
 class ProxyAutomation(BaseAutomation):
+    def get_subgroups(self):
+        conn = db.engine.connect()
+        result = conn.execute("""
+        SELECT origin.group_id, proxy.psg, COUNT(proxy.id) FROM proxy, origin
+        WHERE proxy.origin_id = origin.id
+            AND proxy.destroyed IS NULL
+            AND proxy.provider = :provider
+        GROUP BY origin.group_id, proxy.psg;
+        """, provider=self.provider)
+        subgroups = defaultdict(lambda: defaultdict(lambda: 0))
+        for row in result:
+            subgroups[row[0]][row[1]] = row[2]
+        return subgroups
+
     def create_missing_proxies(self):
         origins = Origin.query.all()
         for origin in origins:
@@ -44,6 +59,7 @@ class ProxyAutomation(BaseAutomation):
                 Proxy.provider == self.provider,
                 Proxy.destroyed == None
             ).all(),
+            subgroups = self.get_subgroups(),
             global_namespace=app.config['GLOBAL_NAMESPACE'],
             bypass_token=app.config['BYPASS_TOKEN'],
             **{