def import_command(import_all, force, path):
"""
Import documents from Lingotek Cloud, automatically downloading to the project root
"""
# todo import should show all documents
# add a force option so can import all force -- overwrites all existing documents without prompting
# check if doc id
# if exist, prompt for overwrite
# else automatically re-name
# possibly have to patch title in Lingotek Cloud?
try:
# action = actions.Action(os.getcwd())
action = ImportAction(os.getcwd())
init_logger(action.path)
action.import_action(import_all, force, path)
except(UninitializedError, RequestFailedError) as e:
print_log(e)
logger.error(e)
return
class TestImport(unittest.TestCase):
@classmethod
def setUpClass(cls):
create_config()
@classmethod
def tearDownClass(cls):
cleanup()
def setUp(self):
self.action = ImportAction(os.getcwd())
self.action.clean_action(False, False, None)
self.files = ['sample.txt', 'sample1.txt', 'sample2.txt']
for fn in self.files:
create_txt_file(fn)
self.doc_ids = []
for fn in self.files:
title = os.path.basename(os.path.normpath(fn))
response = self.action.api.add_document(self.action.locale, fn, self.action.project_id, title)
assert response.status_code == 202
self.doc_ids.append(response.json()['properties']['id'])
for doc_id in self.doc_ids:
assert poll_doc(self.action, doc_id)
for fn in self.files:
delete_file(fn)
self.imported = []
def tearDown(self):
for doc_id in self.doc_ids:
self.action.rm_action(doc_id, id=True)
self.action.clean_action(True, False, None)
self.action.close()
def test_import_all(self):
self.action.import_action(True, True, None)
for doc_id in self.doc_ids:
doc = self.action.doc_manager.get_doc_by_prop('id', doc_id)
assert doc
self.imported.append(doc['id'])
def test_import_locale(self):
locale = "ja_JP"
doc_id = self.doc_ids[0]
response = self.action.api.document_add_target(doc_id, locale)
assert response.status_code == 201
self.action.import_action(False, True, None, doc_id)
entry = self.action.doc_manager.get_doc_by_prop("id", doc_id)
assert locale in entry["locales"]
self.imported.append(entry['id'])
def test_import_no_locale(self):
self.action.import_action(False, True, None, self.doc_ids[0])
entry = self.action.doc_manager.get_doc_by_prop("id", self.doc_ids[0])
assert not entry.get("locales")
self.imported.append(entry['id'])