Skip to content

Commit

Permalink
[0.3.4] Added TM commands and Jobs edit
Browse files Browse the repository at this point in the history
  • Loading branch information
zerodayz committed Mar 27, 2020
1 parent 2ed2f0a commit 201a427
Show file tree
Hide file tree
Showing 10 changed files with 488 additions and 7 deletions.
4 changes: 3 additions & 1 deletion memsource_cli/api/job_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -857,7 +857,7 @@ def edit_parts_with_http_info(self, project_uid, **kwargs): # noqa: E501
returns the request thread.
"""

all_params = ['project_uid', 'body'] # noqa: E501
all_params = ['project_uid', 'body', 'token'] # noqa: E501
all_params.append('async_req')
all_params.append('_return_http_data_only')
all_params.append('_preload_content')
Expand All @@ -884,6 +884,8 @@ def edit_parts_with_http_info(self, project_uid, **kwargs): # noqa: E501
path_params['projectUid'] = params['project_uid'] # noqa: E501

query_params = []
if 'token' in params:
query_params.append(('token', params['token'])) # noqa: E501

header_params = {}

Expand Down
14 changes: 11 additions & 3 deletions memsource_cli/api/translation_memory_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -663,7 +663,7 @@ def download_search_result_with_http_info(self, async_request_id, **kwargs): #
returns the request thread.
"""

all_params = ['async_request_id', 'format'] # noqa: E501
all_params = ['async_request_id', 'format', 'token'] # noqa: E501
all_params.append('async_req')
all_params.append('_return_http_data_only')
all_params.append('_preload_content')
Expand Down Expand Up @@ -692,6 +692,8 @@ def download_search_result_with_http_info(self, async_request_id, **kwargs): #
query_params = []
if 'format' in params:
query_params.append(('format', params['format'])) # noqa: E501
if 'token' in params:
query_params.append(('token', params['token'])) # noqa: E501

header_params = {}

Expand Down Expand Up @@ -867,7 +869,7 @@ def export_with_http_info(self, trans_memory_id, **kwargs): # noqa: E501
returns the request thread.
"""

all_params = ['trans_memory_id', 'format', 'target_lang'] # noqa: E501
all_params = ['trans_memory_id', 'format', 'target_lang', 'token', 'async_req'] # noqa: E501
all_params.append('async_req')
all_params.append('_return_http_data_only')
all_params.append('_preload_content')
Expand Down Expand Up @@ -899,6 +901,10 @@ def export_with_http_info(self, trans_memory_id, **kwargs): # noqa: E501
if 'target_lang' in params:
query_params.append(('targetLang', params['target_lang'])) # noqa: E501
collection_formats['targetLang'] = 'multi' # noqa: E501
if 'token' in params:
query_params.append(('token', params['token'])) # noqa: E501
if 'async_req' in params:
query_params.append(('async_req', params['async_req'])) # noqa: E501

header_params = {}

Expand Down Expand Up @@ -1908,7 +1914,7 @@ def list_trans_memories_with_http_info(self, **kwargs): # noqa: E501
returns the request thread.
"""

all_params = ['name', 'source_lang', 'target_lang', 'client_id', 'domain_id', 'sub_domain_id', 'business_unit_id', 'page_number', 'page_size'] # noqa: E501
all_params = ['name', 'source_lang', 'target_lang', 'client_id', 'domain_id', 'sub_domain_id', 'business_unit_id', 'page_number', 'page_size', 'token'] # noqa: E501
all_params.append('async_req')
all_params.append('_return_http_data_only')
all_params.append('_preload_content')
Expand Down Expand Up @@ -1947,6 +1953,8 @@ def list_trans_memories_with_http_info(self, **kwargs): # noqa: E501
query_params.append(('pageNumber', params['page_number'])) # noqa: E501
if 'page_size' in params:
query_params.append(('pageSize', params['page_size'])) # noqa: E501
if 'token' in params:
query_params.append(('token', params['token'])) # noqa: E501

header_params = {}

Expand Down
72 changes: 69 additions & 3 deletions memsource_cli/job/v1/job.py
Original file line number Diff line number Diff line change
Expand Up @@ -89,6 +89,7 @@ def take_action(self, parsed_args):
return((column_headers), (
(s) for s in data))


class EditJob(ShowOne):
"""
Edit job
Expand All @@ -102,7 +103,7 @@ def get_parser(self, prog_name):
help='project_uid',
dest='project_uid',
required=True
)
)
parser.add_argument(
'--job-id',
help='job_uid',
Expand Down Expand Up @@ -138,14 +139,79 @@ def take_action(self, parsed_args):
provider_type = provider[0]
provider_id = provider[1]
_providers.append({'type': provider_type, 'id': provider_id})


# _jobs = []
# for i in parsed_args.jobs:
# _jobs.append({'uid': i})

response = api.edit_part(token=self.app.client.configuration.token,
project_uid=parsed_args.project_uid,
job_uid=parsed_args.job_uid,
body={"status": parsed_args.status, "providers": _providers, "dateDue": parsed_args.date_due})
return utils._print_output(response)


class EditJobs(ShowOne):
"""
Edit jobs
"""

def get_parser(self, prog_name):
"""Command argument parsing."""
parser = super(EditJobs, self).get_parser(prog_name)
parser.add_argument(
'--project-id',
help='project_uid',
dest='project_uid',
required=True
)
parser.add_argument(
'--jobs',
help='jobs',
dest='jobs',
nargs='+',
default=[]
)
parser.add_argument(
'--status',
help='status',
dest='status',
required=True
)
parser.add_argument(
'--date-due',
help='date_due',
dest='date_due',
)
parser.add_argument(
'--providers',
help='providers',
dest='providers',
nargs='+',
default=[]
)
return parser

def take_action(self, parsed_args):
api = memsource_cli.JobApi(self.app.client)

_providers = []
for i in parsed_args.providers:
provider = i.split('=')
provider_type = provider[0]
provider_id = provider[1]
_providers.append({'type': provider_type, 'id': provider_id})

_jobs = []
for i in parsed_args.jobs:
_jobs.append({'uid': i})

response = api.edit_parts(token=self.app.client.configuration.token,
project_uid=parsed_args.project_uid,
body={"status": parsed_args.status, "providers": _providers, "dateDue": parsed_args.date_due, "jobs": _jobs})
return utils._print_output(response)


class DeleteTranslations(command.Command):
"""
Delete translations
Expand Down Expand Up @@ -352,4 +418,4 @@ def take_action(self, parsed_args):
values = ((parsed_args.type),
(parsed_args.bilingual_format), (file_paths))

return((header), (values))
return((header), (values))
Empty file.
Empty file.
Loading

0 comments on commit 201a427

Please sign in to comment.