1+ import requests
2+ from . import utils
3+ from codat .models import operations
4+ from typing import Optional
5+
6+ class Files :
7+ _client : requests .Session
8+ _security_client : requests .Session
9+ _server_url : str
10+ _language : str
11+ _sdk_version : str
12+ _gen_version : str
13+
14+ def __init__ (self , client : requests .Session , security_client : requests .Session , server_url : str , language : str , sdk_version : str , gen_version : str ) -> None :
15+ self ._client = client
16+ self ._security_client = security_client
17+ self ._server_url = server_url
18+ self ._language = language
19+ self ._sdk_version = sdk_version
20+ self ._gen_version = gen_version
21+
22+
23+ def download_files (self , request : operations .DownloadFilesRequest ) -> operations .DownloadFilesResponse :
24+ r"""Download all files for a company
25+ You can specify a date to download specific files for.
26+ """
27+
28+ base_url = self ._server_url
29+
30+ url = utils .generate_url (base_url , "/companies/{companyId}/files/download" , request .path_params )
31+
32+ query_params = utils .get_query_params (request .query_params )
33+
34+ client = utils .configure_security_client (self ._client , request .security )
35+
36+ r = client .request ("GET" , url , params = query_params )
37+ content_type = r .headers .get ("Content-Type" )
38+
39+ res = operations .DownloadFilesResponse (status_code = r .status_code , content_type = content_type )
40+
41+ if r .status_code == 200 :
42+ pass
43+
44+ return res
45+
46+
47+ def list_files (self , request : operations .ListFilesRequest ) -> operations .ListFilesResponse :
48+ r"""List all files uploaded by a company
49+ Returns an array of files that have been uploaded for a given company.
50+ """
51+
52+ base_url = self ._server_url
53+
54+ url = utils .generate_url (base_url , "/companies/{companyId}/files" , request .path_params )
55+
56+
57+ client = utils .configure_security_client (self ._client , request .security )
58+
59+ r = client .request ("GET" , url )
60+ content_type = r .headers .get ("Content-Type" )
61+
62+ res = operations .ListFilesResponse (status_code = r .status_code , content_type = content_type )
63+
64+ if r .status_code == 200 :
65+ if utils .match_content_type (content_type , "application/json" ):
66+ out = utils .unmarshal_json (r .text , Optional [list [operations .ListFilesFile ]])
67+ res .files = out
68+
69+ return res
70+
71+
72+ def upload_files (self , request : operations .UploadFilesRequest ) -> operations .UploadFilesResponse :
73+ r"""Upload files for a company
74+ Upload files
75+ """
76+
77+ base_url = self ._server_url
78+
79+ url = utils .generate_url (base_url , "/companies/{companyId}/connections/{connectionId}/files" , request .path_params )
80+
81+
82+ client = utils .configure_security_client (self ._client , request .security )
83+
84+ r = client .request ("POST" , url )
85+ content_type = r .headers .get ("Content-Type" )
86+
87+ res = operations .UploadFilesResponse (status_code = r .status_code , content_type = content_type )
88+
89+ if r .status_code == 200 :
90+ pass
91+
92+ return res
93+
94+
0 commit comments