auto/terraform: typing hints for base terraform module

This commit is contained in:
Iain Learmonth 2022-05-16 10:08:18 +01:00
parent ccf0ce6a06
commit 51f580a304
4 changed files with 35 additions and 16 deletions

View file

@ -3,6 +3,7 @@ import datetime
import math
import string
import random
from typing import Dict
from sqlalchemy import text
from tldextract import tldextract
@ -17,7 +18,7 @@ from app.terraform.terraform import TerraformAutomation
class ProxyAutomation(TerraformAutomation):
subgroup_max = math.inf
def get_subgroups(self):
def get_subgroups(self) -> Dict[int, Dict[int, int]]:
conn = db.engine.connect()
result = conn.execute(text("""
SELECT origin.group_id, proxy.psg, COUNT(proxy.id) FROM proxy, origin