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_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 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 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_no_document(self): doc = SimpleQuery.get_by_example(collection=self.test_1_col, example_data={ '_key': 'dddd', }) self.assertEqual(doc, None)
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_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 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 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 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
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