Skip to content

Commit 784939b

Browse files
yolanferyyolanfery
committed
Add datasets methods
1 parent 07695ae commit 784939b

8 files changed

Lines changed: 494 additions & 0 deletions

File tree

‎openhexa/graphql/graphql_client/__init__.py‎

Lines changed: 49 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,11 @@
1717
CreateConnectionCreateConnection,
1818
CreateConnectionCreateConnectionConnection,
1919
)
20+
from .create_dataset import (
21+
CreateDataset,
22+
CreateDatasetCreateDataset,
23+
CreateDatasetCreateDatasetDataset,
24+
)
2025
from .create_pipeline import (
2126
CreatePipeline,
2227
CreatePipelineCreatePipeline,
@@ -43,7 +48,25 @@
4348
CreateWorkspaceCreateWorkspaceWorkspace,
4449
CreateWorkspaceCreateWorkspaceWorkspaceCountries,
4550
)
51+
from .dataset import (
52+
Dataset,
53+
DatasetDataset,
54+
DatasetDatasetCreatedBy,
55+
DatasetDatasetPermissions,
56+
DatasetDatasetVersions,
57+
DatasetDatasetVersionsItems,
58+
DatasetDatasetVersionsItemsCreatedBy,
59+
DatasetDatasetWorkspace,
60+
)
61+
from .datasets import (
62+
Datasets,
63+
DatasetsDatasets,
64+
DatasetsDatasetsItems,
65+
DatasetsDatasetsItemsCreatedBy,
66+
DatasetsDatasetsItemsPermissions,
67+
)
4668
from .delete_connection import DeleteConnection, DeleteConnectionDeleteConnection
69+
from .delete_dataset import DeleteDataset, DeleteDatasetDeleteDataset
4770
from .delete_pipeline import DeletePipeline, DeletePipelineDeletePipeline
4871
from .delete_pipeline_template import (
4972
DeletePipelineTemplate,
@@ -349,6 +372,11 @@
349372
UpdateConnectionUpdateConnectionConnection,
350373
UpdateConnectionUpdateConnectionConnectionFields,
351374
)
375+
from .update_dataset import (
376+
UpdateDataset,
377+
UpdateDatasetUpdateDataset,
378+
UpdateDatasetUpdateDatasetDataset,
379+
)
352380
from .update_webapp import UpdateWebapp, UpdateWebappUpdateWebapp
353381
from .update_workspace import (
354382
UpdateWorkspace,
@@ -422,6 +450,9 @@
422450
"CreateConnectionCreateConnectionConnection",
423451
"CreateConnectionError",
424452
"CreateConnectionInput",
453+
"CreateDataset",
454+
"CreateDatasetCreateDataset",
455+
"CreateDatasetCreateDatasetDataset",
425456
"CreateDatasetError",
426457
"CreateDatasetInput",
427458
"CreateDatasetVersionError",
@@ -465,6 +496,19 @@
465496
"DHIS2ConnectionError",
466497
"DHIS2ConnectionStatus",
467498
"DHIS2MetadataType",
499+
"Dataset",
500+
"DatasetDataset",
501+
"DatasetDatasetCreatedBy",
502+
"DatasetDatasetPermissions",
503+
"DatasetDatasetVersions",
504+
"DatasetDatasetVersionsItems",
505+
"DatasetDatasetVersionsItemsCreatedBy",
506+
"DatasetDatasetWorkspace",
507+
"Datasets",
508+
"DatasetsDatasets",
509+
"DatasetsDatasetsItems",
510+
"DatasetsDatasetsItemsCreatedBy",
511+
"DatasetsDatasetsItemsPermissions",
468512
"DeclineWorkspaceInvitationError",
469513
"DeclineWorkspaceInvitationInput",
470514
"DeleteAccessmodAnalysisError",
@@ -481,6 +525,8 @@
481525
"DeleteConnectionDeleteConnection",
482526
"DeleteConnectionError",
483527
"DeleteConnectionInput",
528+
"DeleteDataset",
529+
"DeleteDatasetDeleteDataset",
484530
"DeleteDatasetError",
485531
"DeleteDatasetInput",
486532
"DeleteDatasetLinkError",
@@ -657,8 +703,11 @@
657703
"UpdateConnectionUpdateConnectionConnectionFields",
658704
"UpdateDAGError",
659705
"UpdateDAGInput",
706+
"UpdateDataset",
660707
"UpdateDatasetError",
661708
"UpdateDatasetInput",
709+
"UpdateDatasetUpdateDataset",
710+
"UpdateDatasetUpdateDatasetDataset",
662711
"UpdateDatasetVersionError",
663712
"UpdateDatasetVersionInput",
664713
"UpdateMembershipError",

‎openhexa/graphql/graphql_client/client.py‎

Lines changed: 168 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
from .base_client import BaseClient
1616
from .base_model import UNSET, UnsetType
1717
from .create_connection import CreateConnection, CreateConnectionCreateConnection
18+
from .create_dataset import CreateDataset, CreateDatasetCreateDataset
1819
from .create_pipeline import CreatePipeline, CreatePipelineCreatePipeline
1920
from .create_pipeline_from_template_version import (
2021
CreatePipelineFromTemplateVersion,
@@ -26,7 +27,10 @@
2627
)
2728
from .create_webapp import CreateWebapp, CreateWebappCreateWebapp
2829
from .create_workspace import CreateWorkspace, CreateWorkspaceCreateWorkspace
30+
from .dataset import Dataset, DatasetDataset
31+
from .datasets import Datasets, DatasetsDatasets
2932
from .delete_connection import DeleteConnection, DeleteConnectionDeleteConnection
33+
from .delete_dataset import DeleteDataset, DeleteDatasetDeleteDataset
3034
from .delete_pipeline import DeletePipeline, DeletePipelineDeletePipeline
3135
from .delete_pipeline_template import (
3236
DeletePipelineTemplate,
@@ -42,13 +46,15 @@
4246
AddToFavoritesInput,
4347
ArchiveWorkspaceInput,
4448
CreateConnectionInput,
49+
CreateDatasetInput,
4550
CreatePipelineFromTemplateVersionInput,
4651
CreatePipelineInput,
4752
CreatePipelineRecipientInput,
4853
CreatePipelineTemplateVersionInput,
4954
CreateWebappInput,
5055
CreateWorkspaceInput,
5156
DeleteConnectionInput,
57+
DeleteDatasetInput,
5258
DeletePipelineInput,
5359
DeletePipelineTemplateInput,
5460
DeletePipelineVersionInput,
@@ -57,6 +63,7 @@
5763
RemoveFromFavoritesInput,
5864
StopPipelineInput,
5965
UpdateConnectionInput,
66+
UpdateDatasetInput,
6067
UpdateWebappInput,
6168
UpdateWorkspaceInput,
6269
UpgradePipelineVersionFromTemplateInput,
@@ -75,6 +82,7 @@
7582
)
7683
from .stop_pipeline import StopPipeline, StopPipelineStopPipeline
7784
from .update_connection import UpdateConnection, UpdateConnectionUpdateConnection
85+
from .update_dataset import UpdateDataset, UpdateDatasetUpdateDataset
7886
from .update_webapp import UpdateWebapp, UpdateWebappUpdateWebapp
7987
from .update_workspace import UpdateWorkspace, UpdateWorkspaceUpdateWorkspace
8088
from .upgrade_pipeline_version_from_template import (
@@ -870,3 +878,163 @@ def get_users(
870878
)
871879
data = self.get_data(response)
872880
return GetUsers.model_validate(data).users
881+
882+
def datasets(
883+
self,
884+
query: Union[Optional[str], UnsetType] = UNSET,
885+
page: Union[Optional[int], UnsetType] = UNSET,
886+
per_page: Union[Optional[int], UnsetType] = UNSET,
887+
**kwargs: Any
888+
) -> DatasetsDatasets:
889+
_query = gql(
890+
"""
891+
query Datasets($query: String, $page: Int = 1, $perPage: Int = 15) {
892+
datasets(query: $query, page: $page, perPage: $perPage) {
893+
totalPages
894+
items {
895+
id
896+
slug
897+
name
898+
description
899+
createdAt
900+
updatedAt
901+
createdBy {
902+
id
903+
displayName
904+
}
905+
permissions {
906+
update
907+
delete
908+
}
909+
}
910+
}
911+
}
912+
"""
913+
)
914+
variables: Dict[str, object] = {
915+
"query": query,
916+
"page": page,
917+
"perPage": per_page,
918+
}
919+
response = self.execute(
920+
query=_query, operation_name="Datasets", variables=variables, **kwargs
921+
)
922+
data = self.get_data(response)
923+
return Datasets.model_validate(data).datasets
924+
925+
def dataset(self, id: str, **kwargs: Any) -> Optional[DatasetDataset]:
926+
query = gql(
927+
"""
928+
query Dataset($id: ID!) {
929+
dataset(id: $id) {
930+
id
931+
slug
932+
name
933+
description
934+
createdAt
935+
updatedAt
936+
createdBy {
937+
id
938+
displayName
939+
email
940+
}
941+
permissions {
942+
update
943+
delete
944+
createVersion
945+
}
946+
versions {
947+
items {
948+
id
949+
name
950+
changelog
951+
createdAt
952+
createdBy {
953+
id
954+
displayName
955+
}
956+
}
957+
}
958+
workspace {
959+
slug
960+
name
961+
}
962+
}
963+
}
964+
"""
965+
)
966+
variables: Dict[str, object] = {"id": id}
967+
response = self.execute(
968+
query=query, operation_name="Dataset", variables=variables, **kwargs
969+
)
970+
data = self.get_data(response)
971+
return Dataset.model_validate(data).dataset
972+
973+
def create_dataset(
974+
self, input: CreateDatasetInput, **kwargs: Any
975+
) -> CreateDatasetCreateDataset:
976+
query = gql(
977+
"""
978+
mutation CreateDataset($input: CreateDatasetInput!) {
979+
createDataset(input: $input) {
980+
success
981+
errors
982+
dataset {
983+
slug
984+
name
985+
}
986+
}
987+
}
988+
"""
989+
)
990+
variables: Dict[str, object] = {"input": input}
991+
response = self.execute(
992+
query=query, operation_name="CreateDataset", variables=variables, **kwargs
993+
)
994+
data = self.get_data(response)
995+
return CreateDataset.model_validate(data).create_dataset
996+
997+
def update_dataset(
998+
self, input: UpdateDatasetInput, **kwargs: Any
999+
) -> UpdateDatasetUpdateDataset:
1000+
query = gql(
1001+
"""
1002+
mutation UpdateDataset($input: UpdateDatasetInput!) {
1003+
updateDataset(input: $input) {
1004+
success
1005+
errors
1006+
dataset {
1007+
slug
1008+
name
1009+
description
1010+
}
1011+
}
1012+
}
1013+
"""
1014+
)
1015+
variables: Dict[str, object] = {"input": input}
1016+
response = self.execute(
1017+
query=query, operation_name="UpdateDataset", variables=variables, **kwargs
1018+
)
1019+
data = self.get_data(response)
1020+
return UpdateDataset.model_validate(data).update_dataset
1021+
1022+
def delete_dataset(
1023+
self, input: DeleteDatasetInput, **kwargs: Any
1024+
) -> DeleteDatasetDeleteDataset:
1025+
query = gql(
1026+
"""
1027+
mutation DeleteDataset($input: DeleteDatasetInput!) {
1028+
deleteDataset(input: $input) {
1029+
success
1030+
errors
1031+
}
1032+
}
1033+
"""
1034+
)
1035+
variables: Dict[str, object] = {"input": input}
1036+
response = self.execute(
1037+
query=query, operation_name="DeleteDataset", variables=variables, **kwargs
1038+
)
1039+
data = self.get_data(response)
1040+
return DeleteDataset.model_validate(data).delete_dataset

‎openhexa/graphql/graphql_client/create_dataset.py‎

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
# Generated by ariadne-codegen
2+
# Source: openhexa/graphql/queries.graphql
3+
4+
from typing import List, Optional
5+
6+
from pydantic import Field
7+
8+
from .base_model import BaseModel
9+
from .enums import CreateDatasetError
10+
11+
12+
class CreateDataset(BaseModel):
13+
create_dataset: "CreateDatasetCreateDataset" = Field(alias="createDataset")
14+
15+
16+
class CreateDatasetCreateDataset(BaseModel):
17+
success: bool
18+
errors: List[CreateDatasetError]
19+
dataset: Optional["CreateDatasetCreateDatasetDataset"]
20+
21+
22+
class CreateDatasetCreateDatasetDataset(BaseModel):
23+
slug: str
24+
name: str
25+
26+
27+
CreateDataset.model_rebuild()
28+
CreateDatasetCreateDataset.model_rebuild()

0 commit comments

Comments
 (0)