Skip to content

Commit

Permalink
refactor: Reduce some codes
Browse files Browse the repository at this point in the history
  • Loading branch information
abawchen committed May 13, 2019
1 parent 617eaad commit 4203aa4
Showing 1 changed file with 20 additions and 25 deletions.
45 changes: 20 additions & 25 deletions graphene_mongo/advanced_types.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,37 +11,32 @@ def _resolve_fs_field(field, name, default_value=None):
return getattr(v, name, default_value)


def _resolve_content_type(self, info):
return _resolve_fs_field(self, 'content_type')


def _resolve_md5(self, info):
return _resolve_fs_field(self, 'md5')


def _resolve_chunk_size(self, info):
return _resolve_fs_field(self, 'chunk_size', 0)

class FileFieldType(graphene.ObjectType):

def _resolve_length(self, info):
return _resolve_fs_field(self, 'length', 0)
content_type = graphene.String()
md5 = graphene.String()
chunk_size = graphene.Int()
length = graphene.Int()
data = graphene.String()

def resolve_content_type(self, info):
return _resolve_fs_field(self, 'content_type')

def _resolve_data(self, info):
v = getattr(self.instance, self.key)
data = v.read()
if data is not None:
return base64.b64encode(data)
return None
def resolve_md5(self, info):
return _resolve_fs_field(self, 'md5')

def resolve_chunk_size(self, info):
return _resolve_fs_field(self, 'chunk_size', 0)

class FileFieldType(graphene.ObjectType):
def resolve_length(self, info):
return _resolve_fs_field(self, 'length', 0)

content_type = graphene.String(resolver=_resolve_content_type)
md5 = graphene.String(resolver=_resolve_md5)
chunk_size = graphene.Int(resolver=_resolve_chunk_size)
length = graphene.Int(resolver=_resolve_length)
data = graphene.String(resolver=_resolve_data)
def resolve_data(self, info):
v = getattr(self.instance, self.key)
data = v.read()
if data is not None:
return base64.b64encode(data)
return None


class _TypeField(graphene.ObjectType):
Expand Down

0 comments on commit 4203aa4

Please sign in to comment.