class UserGroupManager(AlchemyManager): model = UserGroup fields = ('user_id', 'group_id') create_fields = ( 'user_id', 'group_id', ) update_fields = tuple() list_fields = ('user_id', 'group_id') _field_validators = { 'user_id': StringField('user_id', required=True), 'group_id': StringField('group_id', required=True) }
class UserManager(AlchemyManager): model = User fields = ('id', 'username', 'first_name') create_fields = ('username', 'first_name') update_fields = ('first_name', ) _field_validators = { # username must be between 3 and 20 alphanumeric characters 'username': StringField('username', minimum=3, regex=re.compile(r'^[a-zA-Z0-9]{3,20}'), required=True), 'first_name': StringField('first_name', required=True) }
def test_translate_keeps_undefined_fields(self): """ Tests that any fields not in the defined dictionary are still included. """ field_dict = [StringField('field1'), IntegerField('field2')] field = DictField('f', required=True, field_list=field_dict) input_vals = dict(field1='hey', field2='5', field3='Who Cares?') resp = field.translate(input_vals, validate=True) self.assertIn('field3', resp)
def test_translate_success(self): """ Tests the expected conditions for the translate method. """ field_dict = [StringField('field1'), IntegerField('field2')] field = DictField('f', required=True, field_list=field_dict) input_vals = dict(field1='hey', field2='5') resp = field.translate(input_vals, validate=True) self.assertDictEqual(dict(field1='hey', field2=5), resp)
class GroupManager(AlchemyManager): model = Group fields = ('id', 'name', 'users.id', 'users.username', 'users.first_name') create_fields = ('name', ) update_fields = tuple() list_fields = ( 'id', 'name', ) _field_validators = { 'name': StringField('name', minimum=3, maximum=20), 'id': IntegerField('id') }
class HelloWorldViewset(ResourceBase): namespace = name_space manager = MM1() resource_name = 'myresource' _relationships = [ Relationship('related', property_map={'related': 'id'}, relation='ComplimentaryViewset') ] @apimethod(methods=['GET']) @translate(fields=[StringField('content')], validate=True) def hello(cls, request, *args, **kwargs): return cls(properties=request.query_args)
class PostManager(AlchemyManager): model = Post fields = ( 'id', 'username', 'post_text', 'comments.id', ) create_fields = ( 'username', 'post_text', ) _field_validators = { 'username': StringField('username', minimum=3, regex=re.compile(r'^[a-zA-Z0-9]+$')) }
class ItemManager(BaseManager): """Manager that interops between ripozo and grpc. Provides convience functions primarily for basic CRUD. """ fields = ( 'id', 'name', ) field_validators = [ IntegerField('id', required=True), StringField('name', required=True), ] def create(self, values, *args, **kwargs): raise NotImplementedError() def delete(self, lookup_keys, *args, **kwargs): raise NotImplementedError() def retrieve(self, lookup_keys, *args, **kwargs): """Retrieve a single item and nothing more as a dictionary. Args: lookup_keys (dict): Taken from url_params on flask request. Used to lookup a item and its associated values. Returns: dict or None: Properties of the retrieved item or None if no such item found. """ resp = items_conn.stub.GetItem( items_pb2.GetItemRequest(id=lookup_keys['id'])) item = resp.item if item: return dict(id=item.id, name=item.name) def retrieve_list(self, filters, *args, **kwargs): raise NotImplementedError() def update(self, lookup_keys, updates, *args, **kwargs): raise NotImplementedError()
def test_translate_subfield_failure(self): """ Tests that when one of the child fields fails then this field as a whole will fail. """ field_dict = [ StringField('field1'), IntegerField('field2', maximum=10) ] field = DictField('f', required=True, field_list=field_dict) input_vals = dict(field1='hey', field2='notanumber') self.assertRaises(TranslationException, field.translate, input_vals, validate=True) input_vals = dict(field1='hey', field2='11') self.assertRaises(ValidationException, field.translate, input_vals, validate=True)
def test_validate_regex(self): f = StringField('field', regex=re.compile(r'^something$')) self.assertRaises(ValidationException, f._validate, 'notsomething') original = 'something' new = f._validate(original) self.assertEqual(original, new)
def test_validate_regex(self): f = StringField('field', regex=re.compile(r'^something$')) self.assertRaises(ValidationException, f._validate, 'notsomething') original = 'something' new = f._validate(original) self.assertEqual(original, new)