python-tfe/src/pytfe/models/policy.py at ccf12340db04b4de5e252d5fbfde9cc72ebbe9ef · hashicorp/python-tfe · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
from __future__ import annotations

from datetime import datetime

from pydantic import BaseModel, ConfigDict, Field

from .organization import Organization
from .policy_types import EnforcementLevel, PolicyKind


class Policy(BaseModel):
    model_config = ConfigDict(populate_by_name=True, validate_by_name=True)

    id: str
    name: str | None = Field(None, alias="name")
    kind: PolicyKind | None = Field(None, alias="kind")
    query: str | None = Field(None, alias="query")
    description: str | None = Field(None, alias="description")
    enforcement_level: EnforcementLevel | None = Field(None, alias="enforcement-level")
    policy_set_count: int | None = Field(None, alias="policy-set-count")
    updated_at: datetime | None = Field(None, alias="updated-at")
    organization: Organization | None = Field(None, alias="organization")


class PolicyList(BaseModel):
    model_config = ConfigDict(populate_by_name=True, validate_by_name=True)

    items: list[Policy] = Field(default_factory=list)
    current_page: int | None = None
    total_pages: int | None = None
    prev_page: int | None = None
    next_page: int | None = None
    total_count: int | None = None


class PolicyListOptions(BaseModel):
    model_config = ConfigDict(populate_by_name=True, validate_by_name=True)

    search: str | None = Field(None, alias="search[name]")
    kind: PolicyKind | None = Field(None, alias="filter[kind]")
    page_number: int | None = Field(None, alias="page[number]")
    page_size: int | None = Field(None, alias="page[size]")


class PolicyCreateOptions(BaseModel):
    model_config = ConfigDict(populate_by_name=True, validate_by_name=True)

    name: str = Field(..., alias="name")
    kind: PolicyKind | None = Field(None, alias="kind")
    query: str | None = Field(None, alias="query")
    description: str | None = Field(None, alias="description")
    enforcement_level: EnforcementLevel | None = Field(None, alias="enforcement-level")


class PolicyUpdateOptions(BaseModel):
    model_config = ConfigDict(populate_by_name=True, validate_by_name=True)

    query: str | None = Field(None, alias="query")
    description: str | None = Field(None, alias="description")
    enforcement_level: EnforcementLevel | None = Field(None, alias="enforcement-level")