Skip to content

Commit 6ba2db3

Browse files
committed
refactor: Move _resolve_fs_field into FileFieldType as a classmethod
1 parent 64cca49 commit 6ba2db3

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

graphene_mongo/advanced_types.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -2,11 +2,6 @@
22
import graphene
33

44

5-
def _resolve_fs_field(field, name, default_value=None):
6-
v = getattr(field.instance, field.key)
7-
return getattr(v, name, default_value)
8-
9-
105
class FileFieldType(graphene.ObjectType):
116

127
content_type = graphene.String()
@@ -15,17 +10,22 @@ class FileFieldType(graphene.ObjectType):
1510
length = graphene.Int()
1611
data = graphene.String()
1712

13+
@classmethod
14+
def _resolve_fs_field(cls, field, name, default_value=None):
15+
v = getattr(field.instance, field.key)
16+
return getattr(v, name, default_value)
17+
1818
def resolve_content_type(self, info):
19-
return _resolve_fs_field(self, 'content_type')
19+
return FileFieldType._resolve_fs_field(self, 'content_type')
2020

2121
def resolve_md5(self, info):
22-
return _resolve_fs_field(self, 'md5')
22+
return FileFieldType._resolve_fs_field(self, 'md5')
2323

2424
def resolve_chunk_size(self, info):
25-
return _resolve_fs_field(self, 'chunk_size', 0)
25+
return FileFieldType._resolve_fs_field(self, 'chunk_size', 0)
2626

2727
def resolve_length(self, info):
28-
return _resolve_fs_field(self, 'length', 0)
28+
return FileFieldType._resolve_fs_field(self, 'length', 0)
2929

3030
def resolve_data(self, info):
3131
v = getattr(self.instance, self.key)

0 commit comments

Comments
 (0)