async def index( dns_server: str, sort_qs: SortQS = Depends(SortQS), search: str = Query(None), pagination: PaginationQS = Depends(PaginationQS), zone_repo: ZoneRepo = Depends(ZoneRepo), token: TokenPayload = Depends(ScopedTo("zone:list")), includes: List[str] = Query(None), ): includes = only(includes, ["zones"], values=True) # Support ability to either submit dns_server.name or dns_server.id as dns_server_id zone_dns_server_id_label = zone_repo.label("dns_server_id") try: dns_server = int(dns_server) label = zone_dns_server_id_label except ValueError: label = zone_repo.label("dns_server.name") pg, items = (zone_repo.search(search).loads(includes).filter_or( label == dns_server, zone_dns_server_id_label.is_(None)).sort( sort_qs).paginate(pagination).includes(includes).data()) return ZonesResponse(pagination=pg, zones=items)
async def store( form: DnsRequestCreateForm, dns_request_repo: DnsRequestRepo = Depends(DnsRequestRepo()), zone_repo: ZoneRepo = Depends(ZoneRepo()), dns_server_repo: DnsServerRepo = Depends(DnsServerRepo()), token: str = Depends(ScopedTo("dns-request:create")), ): dns_server_id = (dns_server_repo.first_or_fail( name=form.dns_server_name.lower()).results().id) zone = (zone_repo.filter( literal(form.name.lower()).contains( zone_repo.label("domain"))).first().results()) zone_id = zone.id if zone else None data = only(dict(form), ["name", "source_address", "source_port", "type", "protocol"]) data["name"] = data["name"].lower() data["type"] = data["type"].upper() data["dns_server_id"] = dns_server_id data["zone_id"] = zone_id dns_request = dns_request_repo.create(data).data() return DnsRequestResponse(dns_request=dns_request)