Return a custom class for sized InputFile instead extra attrs

This commit is contained in:
Lonami Exo
2018-01-18 20:09:59 +01:00
parent 0e4611a593
commit b546c02210
3 changed files with 24 additions and 18 deletions
+1
View File
@@ -1,2 +1,3 @@
from .draft import Draft
from .dialog import Dialog
from .input_sized_file import InputSizedFile
+9
View File
@@ -0,0 +1,9 @@
from ..types import InputFile
class InputSizedFile(InputFile):
"""InputFile class with two extra parameters: md5 (digest) and size"""
def __init__(self, id_, parts, name, md5, size):
super().__init__(id_, parts, name, md5.hexdigest())
self.md5 = md5.digest()
self.size = size