def create(self, validated_data): #Init Arango Models AwsCredentials.init() Companies.init() account_obj = AwsCredentials.objects.get( _id=str(validated_data['account_id'])) obj = Collection.get_loaded_collection(name='ScheduleIntervalSettings') SimpleQuery.update_by_example( obj, { 'account_id': str(account_obj.document), 'service': str(validated_data['service']), }, { 'service': validated_data['service'], 'repeat_delay': validated_data['repeat_delay'] }) response = SimpleQuery.get_by_example( obj, example_data={ 'account_id': str(account_obj.document), 'service': str(validated_data['service']) }) headers = {"Content-Type": "application/json"} url = FOXX_BASE_URL + "execute/" + urllib.quote_plus( str(response)) + "/" + str(self.context['user']) + "/edit" response = requests.get(url, headers=headers) return response
def validate(self, data): collection = Collection.get_loaded_collection(name='AwsCredentials') obj = SimpleQuery.get_by_example(collection, example_data={ 'company': self.context['company'], 'external_id': data['external_id'] }) if obj: raise serializers.ValidationError( "External id already exist for a company") obj = SimpleQuery.get_by_example(collection, example_data={ 'company': self.context['company'], 'role_arn': data['role_arn'] }) if obj: raise serializers.ValidationError("Role Arn already exist") obj = SimpleQuery.get_by_example(collection, example_data={ 'company': self.context['company'], 'account_name': data['account_name'] }) if obj: raise serializers.ValidationError( "Account name already exist for a company") return data
def test_update_document(self): SimpleQuery.update_by_example(collection=self.test_1_col, example_data={'bla': 'xxx'}, new_value={'bla': 'ttt'}) self.col1_doc2.retrieve() self.assertEqual(self.col1_doc2.bla, 'ttt')
def test_update_document(self): SimpleQuery.update_by_example(collection=self.test_1_col, example_data={ 'bla': 'xxx' }, new_value={ 'bla': 'ttt' }) self.col1_doc2.retrieve() self.assertEqual(self.col1_doc2.bla, 'ttt')
def test_replace_document(self): SimpleQuery.replace_by_example(collection=self.test_1_col, example_data={ 'bla': 'xxx' }, new_value={ 'test': 'foo' }) self.col1_doc2.retrieve() self.assertEqual(self.col1_doc2.test, 'foo')
def test_replace_document(self): SimpleQuery.replace_by_example(collection=self.test_1_col, example_data={'bla': 'xxx'}, new_value={'test': 'foo'}) self.col1_doc2.retrieve() self.assertEqual(self.col1_doc2.test, 'foo')
def test_remove_document(self): SimpleQuery.remove_by_example(collection=self.test_1_col, example_data={ 'bla': 'xxx' }) all_docs = self.test_1_col.documents() self.assertEqual(len(all_docs), 1) doc = all_docs[0] self.assertDocumentsEqual(doc, self.col1_doc1)
def test_remove_document(self): SimpleQuery.remove_by_example(collection=self.test_1_col, example_data={'bla': 'xxx'}) all_docs = self.test_1_col.documents() self.assertEqual(len(all_docs), 1) doc = all_docs[0] self.assertDocumentsEqual(doc, self.col1_doc1)
def test_get_no_document(self): doc = SimpleQuery.get_by_example(collection=self.test_1_col, example_data={ '_key': 'dddd', }) self.assertEqual(doc, None)
def test_create_document(self): trans = Transaction(collections={ 'write': [ self.operating_collection, ] }) # Uses already chosen database as usual collection = trans.collection(name=self.operating_collection) collection.create_document(data={ 'test': 'foo' }) ctrl = TransactionController() transaction_result = ctrl.start(transaction=trans) transaction_doc = create_document_from_result_dict(transaction_result['result'], self.test_1_col.api) created_doc = SimpleQuery.get_by_example(self.test_1_col, example_data={ '_id': transaction_doc.id }) self.assertDocumentsEqual(transaction_doc, created_doc)
def test_get_document_by_example(self): uid = self.col1_doc1.key doc = SimpleQuery.get_by_example(collection=self.test_1_col, example_data={ '_key': uid, }) self.assertDocumentsEqual(doc, self.col1_doc1)
def test_get_document_by_example(self): uid = self.col1_doc1.key doc = SimpleQuery.get_by_example(collection=self.test_1_col, example_data={ '_key': uid, }) self.assertDocumentsEqual(doc, self.col1_doc1)
def company(self): collection = Collection.get_loaded_collection(name='UserCompanies') obj = SimpleQuery.get_by_example(collection, example_data={ 'user': int(self.id), 'active': True }) return obj.company
def test_get_all_documents(self): docs = SimpleQuery.all(collection=self.test_1_col) self.assertEqual(len(docs), 2) doc1 = docs[0] doc2 = docs[1] self.assertNotEqual(doc1, doc2)
def test_get_all_documents(self): docs = SimpleQuery.all(collection=self.test_1_col) self.assertEqual(len(docs), 2) doc1 = docs[0] doc2 = docs[1] self.assertNotEqual(doc1, doc2)
def create(self, validated_data): #Init Arango Models AwsCredentials.init() Companies.init() obj = AwsCredentials() obj.role_arn = str(validated_data['role_arn']) obj.external_id = str(validated_data['external_id']) obj.account_name = str(validated_data['account_name']) obj.company = Companies.objects.get(_id=str(self.context['company'])) obj.user = int(self.context['user']) obj.save() collection = Collection.get_loaded_collection('AwsExternalIds') SimpleQuery.update_by_example(collection, {'external_id': obj.external_id}, { 'used': True, 'company': str(obj.company.id), 'user': obj.user }) return obj.id
def get(self, **kwargs): """ """ collection = self._model_class.collection_instance doc = SimpleQuery.get_by_example(collection=collection, example_data=kwargs) if doc is None: return None model = self._create_model_from_doc(doc=doc) return model
def get(self, **kwargs): """ """ collection = self._model_class.collection_instance doc = SimpleQuery.get_by_example(collection=collection, example_data=kwargs) if doc is None: return None model = self._create_model_from_doc(doc=doc) return model
def test_int32_doc_save(self): for x in xrange(300): doc = self.col1.create_document() doc.int32 = 2**32 doc.int32_1 = 2**32 - 1 doc.int31 = 2**31 doc.int31_1 = 2**31 - 1 doc.save() docs = SimpleQuery.all(self.col1) self.assertEqual(len(docs),300) for x in xrange(len(docs)): self.assertEqual(docs[x].int32,2**32) self.assertEqual(docs[x].int32_1,2**32 - 1) self.assertEqual(docs[x].int31,2**31) self.assertEqual(docs[x].int31_1,2**31 - 1)
def test_dec10_save(self): for i in xrange(100): doc = self.col3.create_document() doc.dec10 = 10**10; doc.dec10_1 = 10**10 - 1 doc.dec9 = 10**9 doc.dec9_1 = 10**9 - 1 doc.save() docs = SimpleQuery.all(self.col3) self.assertEqual(len(docs),100) for j in xrange(len(docs)): self.assertEqual(docs[j].dec10,10**10) self.assertEqual(docs[j].dec10_1,10**10 - 1) self.assertEqual(docs[j].dec9,10**9) self.assertEqual(docs[j].dec9_1,10**9 - 1)
def test_dec19_save(self): for i in xrange(100): doc = self.col4.create_document() doc.dec18 = 10**15; doc.dec18_1 = 10**15 - 1 doc.dec17 = 10**14 doc.dec17_1 = 10**14 - 1 doc.save() docs = SimpleQuery.all(self.col4) self.assertEqual(len(docs),100) for j in xrange(len(docs)): self.assertEqual(docs[j].dec18,10**15) self.assertEqual(docs[j].dec18_1,10**15 - 1) self.assertEqual(docs[j].dec17,10**14) self.assertEqual(docs[j].dec17_1,10**14 - 1)
def test_int64_doc_save(self): for y in xrange(300): doc = self.col2.create_document() doc.int62 = 2**52 doc.int62_1 = 2**52 - 1 doc.int63 = 2**53 doc.int63_1 = 2**53 - 1 doc.save() docs = SimpleQuery.all(self.col2) self.assertEqual(len(docs),300) for y in xrange(len(docs)): self.assertEqual(docs[y].int62,2**52) self.assertEqual(docs[y].int62_1,2**52 - 1) self.assertEqual(docs[y].int63,2**53) self.assertEqual(docs[y].int63_1,2**53 - 1)
def get(self, request, format='json'): """ API View that receives a GET request. and will return a single random unused external id. """ obj = Collection.get_loaded_collection(name='AwsExternalIds') document = SimpleQuery.get_by_example(obj, example_data={ 'used': False, }, allow_multiple=True) if document: external_id = Document(random.choice(document), '', 'AwsExternalIds', client.api).retrieve()['external_id'] else: external_id = "Not found" return Response(external_id, status=status.HTTP_201_CREATED)
def validate(self, data): collection = Collection.get_loaded_collection( name='ScheduleIntervalSettings') obj = SimpleQuery.get_by_example(collection, example_data={ 'account_id': str(data['account_id']), 'service': str(data['service']), }) if not obj: raise serializers.ValidationError( "A setting for the given service with this account not exist") AwsCredentials.init() Companies.init() aws_obj = AwsCredentials.objects.get(_id=str(data['account_id'])) if str(aws_obj.company.id) != str(self.context['company']): raise serializers.ValidationError( 'Companies account are different for Aws and currenly logged in User' ) return data
def test_create_document(self): trans = Transaction( collections={'write': [ self.operating_collection, ]}) # Uses already chosen database as usual collection = trans.collection(name=self.operating_collection) collection.create_document(data={'test': 'foo'}) ctrl = TransactionController() transaction_result = ctrl.start(transaction=trans) transaction_doc = create_document_from_result_dict( transaction_result['result'], self.test_1_col.api) created_doc = SimpleQuery.get_by_example( self.test_1_col, example_data={'_id': transaction_doc.id}) self.assertDocumentsEqual(transaction_doc, created_doc)
def validate_company(self, data): collection = Collection.get_loaded_collection(name='Companies') if SimpleQuery.get_by_example(collection, example_data= {'name': str(data)}): raise serializers.ValidationError("Company already exist") return data
def retrieve_simply_documents(big_collection): """ """ SimpleQuery.all(collection=big_collection)
def test_get_no_document(self): doc = SimpleQuery.get_by_example(collection=self.test_1_col, example_data={ '_key': 'dddd', }) self.assertEqual(doc, None)
def get_default_role(): obj = Collection.get_loaded_collection(name='Roles') return SimpleQuery.get_by_example(obj, example_data={'slug': 'user'})
def create(self, validated_data): user = User.objects.create_user( username=validated_data['email'], password=validated_data['password'], email=validated_data['email'], first_name=validated_data['first_name'], last_name=validated_data['last_name'] if 'last_name' in validated_data else '', is_active=False ) try: # Saving data into UserProfile collection UserProfiles.init() obj = UserProfiles() obj.user = int(user.id) obj.country = str(validated_data['country']) obj.state = str(validated_data['state']) obj.phone_number = str(validated_data['phone_number']) obj.role = str(get_default_role().id) obj.save() # Saving company into collection Companies.init() obj = Companies() obj.name = str(validated_data['company']) obj.save() # Assign User to Company collection = Collection.get_loaded_collection(name='UserCompanies') doc = collection.create_document() doc.user = int(user.id) doc.company = str(obj.id) doc.active = True doc.created_on = str(datetime.datetime.now()) doc.save() except Exception as e: # Removing data saved into document if any error occured collection = Collection.get_loaded_collection(name='UserProfiles') obj = SimpleQuery.get_by_example(collection, example_data={ 'user': int(user.id), }) if obj: Document(obj.id, '', 'UserProfiles', client.api).delete() collection = Collection.get_loaded_collection(name='Companies') obj = SimpleQuery.get_by_example(collection, example_data={ 'name': str(validated_data['company']), }) if obj: Document(obj.id, '', 'Companies', client.api).delete() collection = Collection.get_loaded_collection(name='UserCompanies') obj = SimpleQuery.get_by_example(collection, example_data={ 'user': int(user.id), 'company': str(obj.id) }) if obj: Document(obj.id, '', 'UserCompanies', client.api).delete() raise Exception('Error Occured '+str(e)) return user