def get_data_hs(code, field, field_value): # TODO: try-except obj = Dictionary(code) obj_names = DictionaryNames() document = obj.get_document({str(field): field_value}) origin_dict = obj_names.get_by_code(code) if 'oid' in origin_dict: # Работаем с НСИ справочником data = document oid = origin_dict['oid'] else: try: try: linked_dict = _get_linked_dict(document, origin_dict) except AttributeError: raise InvalidAPIUsage(u'Not found', status_code=404) except KeyError: raise InvalidAPIUsage(u'Not found', status_code=404) if not document: return make_response( vesta_jsonify( dict(oid=linked_dict['oid'], message="not found")), 404) except TypeError, e: raise InvalidAPIUsage(e.message, status_code=400) except InvalidId, e: raise InvalidAPIUsage(e.message, status_code=404)
def document_details(self, code, document_id): obj = Dictionary(code) #TODO: учесть auth_token try: result = obj.get_document(dict(_id=document_id)) except TypeError, e: raise InvalidAPIUsage(e.message, status_code=400)
def document_by_field(self, code, field, field_value): obj = Dictionary(code) if field == 'id': field_value = int(field_value) try: result = obj.get_document({str(field): field_value}) except TypeError, e: raise InvalidAPIUsage(e.message, status_code=400)
def doc_edit(code, _id): dict_names = DictionaryNames() current = dict_names.get_by_code(code) linked = None linked_code = None linked_docs = [] linked_dict = dict() linked_dicts = dict_names.get_list({'version': {'$exists': True}}) obj = Dictionary(code) document = obj.get_document({'_id': _id}) linked_dict = _get_linked_dict(document, current) if linked_dict: linked_code = linked_dict['code'] linked = Dictionary(linked_code) linked_docs = linked.get_list() if request.method == 'POST': data = dict() if request.form: for key, value in request.form.items(): data[key] = value linked_id = data.pop('linked') _post_linked_code = data.pop('linked_collection') data.update({'_id': _id}) if _post_linked_code: if _post_linked_code != linked_code: linked_code = _post_linked_code data.update({'linked_collection': linked_code}) else: obj.unset_attr(_id, 'linked_collection') if linked_id: linked = Dictionary(linked_code) linked_document = linked.get_document({'_id': linked_id}) data.update({linked_code: linked_document}) else: obj.unset_attr(_id, linked_code) obj.unset_attr(_id, 'linked_collection') try: obj.add_document(data) except Exception, e: flash(e, 'error') else: flash(u'Документ сохранён', 'info') return redirect(url_for('.doc_edit', code=code, _id=_id))
def find_data_hs(code): data = APIMixin().parse_request(request) obj = Dictionary(code) obj_names = DictionaryNames() try: _dict = obj_names.get_by_code(code) result = obj.get_document(prepare_find_params(data)) except TypeError, e: raise InvalidAPIUsage(e.message, status_code=400)
def _set_cities_parents(cities): obj = Dictionary(CITY_CODE) result = [] for city in cities: city['parents'] = [] identparent = city['identparent'] parent = city.get('parent') if identparent or parent: level = int(city['level']) for i in xrange(level - 1, 0, -1): if parent: parent_city = obj.get_document({'_id': parent}) elif identparent: parent_city = obj.get_document({'identcode': identparent}) else: break city['parents'].append(parent_city) parent = parent_city.get('parent') identparent = parent_city['identparent'] result.append(city) return result
def get_linked_data(code, field, field_value): # TODO: try-except obj = Dictionary(code) obj_names = DictionaryNames() document = obj.get_document({str(field): field_value}) try: origin_dict = obj_names.get_by_code(code) try: linked_dict = _get_linked_dict(document, origin_dict) except AttributeError: raise InvalidAPIUsage(u'Not found', status_code=404) except KeyError: raise InvalidAPIUsage(u'Not found', status_code=404) if not document: return make_response( vesta_jsonify(dict(oid=linked_dict['oid'], data={})), 200) except TypeError, e: raise InvalidAPIUsage(e.message, status_code=400)
class DictionaryTestCase(unittest.TestCase): def setUp(self): app.config['TESTING'] = True app.config['MONGODB_DB'] = TEST_DB self.app = app.test_client() self.collection = 'test_dictionary' self.obj = Dictionary(self.collection) def drop_collection(self): db.drop_collection(self.collection) def tearDown(self): db.drop_collection(self.collection) def test_get_list_empty(self): self.drop_collection() with self.assertRaises(ValueError): result = self.obj.get_list() #self.assertEqual(result.count(), 0) def test_add_document(self): data = dict(code='test', name=u'Тестовый документ') _id = self.obj.add_document(data) self.assertIsNotNone(_id) result = self.obj.get_document(dict(_id=_id)) data.update(dict(_id=_id)) self.assertDictEqual(data, result) def test_add_documents(self): self.drop_collection() data = [dict(name=u'Документ1', code='document1'), dict(name=u'Документ2'), dict(name=u'Документ3', alias='document2'), dict(name=u'Документ4', code='document4', created=datetime.now())] _ids = self.obj.add_documents(data) self.assertIsNotNone(_ids) self.assertIsInstance(_ids, list) result = list(self.obj.get_list()) self.assertEqual(len(result), len(data)) self.assertEqual(result[0]['code'], data[0]['code']) data_2 = data[2] data_2.update(dict(_id=_ids[2])) self.assertEqual(result[2], data_2) def test_get_document(self): data = dict(code='test5', name=u'Тестовый документ5') _id = self.obj.add_document(data) self.assertIsNotNone(_id) result = self.obj.get_document(dict(_id=_id)) data.update(dict(_id=_id)) self.assertDictEqual(data, result) def test_count_documents(self): self.drop_collection() data = [dict(name=u'Документ1', code='document1'), dict(name=u'Документ2'), dict(name=u'Документ3', alias='document2'), dict(name=u'Документ4', code='document4', created=datetime.now())] _ids = self.obj.add_documents(data) self.assertIsNotNone(_ids) self.assertIsInstance(_ids, list) result = self.obj.count() self.assertEqual(result, len(data)) result = self.obj.count(dict(_id=_ids[0])) self.assertEqual(result, 1) result = self.obj.count(dict(_id='not_exists_id')) self.assertEqual(result, 0) def test_delete(self): data = dict(code='test_delete', name=u'Тестовый документ') _id = self.obj.add_document(data) self.assertIsNotNone(_id) self.obj.delete(_id) result = self.obj.get_document(dict(_id=_id)) self.assertIsNone(result) def test_exists(self): self.drop_collection() data = [dict(name=u'Документ1', code='document1'), dict(name=u'Документ2'), dict(name=u'Документ3', alias='document2'), dict(name=u'Документ4', code='document4', created=datetime.now())] _ids = self.obj.add_documents(data) self.assertIsNotNone(_ids) self.assertIsInstance(_ids, list) result = self.obj.exists() self.assertTrue(result) result = self.obj.count(dict(_id=_ids[0])) self.assertTrue(result) result = self.obj.count(dict(_id='not_exists_id')) self.assertFalse(result)