Save size in bytes and compute & save human_size.
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
#!/usr/bin/env python3.6
|
||||
|
||||
from json import dumps
|
||||
from utils import humansize, representsInteger
|
||||
|
||||
RELEASE_TYPES = ('bdremux', 'brremux', 'remux',
|
||||
'bdrip', 'brrip', 'blu-ray', 'bluray', 'bdmv', 'bdr', 'bd5',
|
||||
@@ -15,7 +16,7 @@ RELEASE_TYPES = ('bdremux', 'brremux', 'remux',
|
||||
class Torrent(object):
|
||||
def __init__(self, name, magnet=None, size=None, uploader=None, date=None,
|
||||
seed_count=None, leech_count=None, url=None):
|
||||
self.name = name
|
||||
self.name = name
|
||||
self.magnet = magnet
|
||||
self.size = size
|
||||
self.uploader = uploader
|
||||
@@ -24,6 +25,9 @@ class Torrent(object):
|
||||
self.leech_count = leech_count
|
||||
self.url = url
|
||||
|
||||
if (size != '' and representsInteger(size)):
|
||||
self.human_size = humansize(int(size))
|
||||
|
||||
def find_release_type(self):
|
||||
name = self.name.casefold()
|
||||
return [r_type for r_type in RELEASE_TYPES if r_type in name]
|
||||
|
||||
Reference in New Issue
Block a user