Пример #1
0
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)
Пример #2
0
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)