137 lines
4.7 KiB
Python
Executable File
137 lines
4.7 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import re
|
|
from .patterns import patterns, types
|
|
|
|
|
|
class PTN(object):
|
|
def _escape_regex(self, string):
|
|
return re.sub('[\-\[\]{}()*+?.,\\\^$|#\s]', '\\$&', string)
|
|
|
|
def __init__(self):
|
|
self.torrent = None
|
|
self.excess_raw = None
|
|
self.group_raw = None
|
|
self.start = None
|
|
self.end = None
|
|
self.title_raw = None
|
|
self.parts = None
|
|
|
|
def _part(self, name, match, raw, clean):
|
|
# The main core instructuions
|
|
self.parts[name] = clean
|
|
|
|
if len(match) != 0:
|
|
# The instructions for extracting title
|
|
index = self.torrent['name'].find(match[0])
|
|
if index == 0:
|
|
self.start = len(match[0])
|
|
elif self.end is None or index < self.end:
|
|
self.end = index
|
|
|
|
if name != 'excess':
|
|
# The instructions for adding excess
|
|
if name == 'group':
|
|
self.group_raw = raw
|
|
if raw is not None:
|
|
self.excess_raw = self.excess_raw.replace(raw, '')
|
|
|
|
def _late(self, name, clean):
|
|
if name == 'group':
|
|
self._part(name, [], None, clean)
|
|
elif name == 'episodeName':
|
|
clean = re.sub('[\._]', ' ', clean)
|
|
clean = re.sub('_+$', '', clean)
|
|
self._part(name, [], None, clean.strip())
|
|
|
|
def parse(self, name):
|
|
self.parts = {}
|
|
self.torrent = {'name': name}
|
|
self.excess_raw = name
|
|
self.group_raw = ''
|
|
self.start = 0
|
|
self.end = None
|
|
self.title_raw = None
|
|
|
|
for key, pattern in patterns:
|
|
if key not in ('season', 'episode', 'website'):
|
|
pattern = r'\b%s\b' % pattern
|
|
|
|
clean_name = re.sub('_', ' ', self.torrent['name'])
|
|
match = re.findall(pattern, clean_name, re.I)
|
|
if len(match) == 0:
|
|
continue
|
|
|
|
index = {}
|
|
if isinstance(match[0], tuple):
|
|
match = list(match[0])
|
|
if len(match) > 1:
|
|
index['raw'] = 0
|
|
index['clean'] = 1
|
|
else:
|
|
index['raw'] = 0
|
|
index['clean'] = 0
|
|
|
|
if key in types.keys() and types[key] == 'boolean':
|
|
clean = True
|
|
else:
|
|
clean = match[index['clean']]
|
|
if key in types.keys() and types[key] == 'integer':
|
|
clean = int(clean)
|
|
if key == 'group':
|
|
if re.search(patterns[5][1], clean, re.I) \
|
|
or re.search(patterns[4][1], clean):
|
|
continue # Codec and quality.
|
|
if re.match('[^ ]+ [^ ]+ .+', clean):
|
|
key = 'episodeName'
|
|
if key == 'episode':
|
|
sub_pattern = self._escape_regex(match[index['raw']])
|
|
self.torrent['map'] = re.sub(
|
|
sub_pattern, '{episode}', self.torrent['name']
|
|
)
|
|
self._part(key, match, match[index['raw']], clean)
|
|
|
|
# Start process for title
|
|
raw = self.torrent['name']
|
|
if self.end is not None:
|
|
raw = raw[self.start:self.end].split('(')[0]
|
|
|
|
clean = re.sub('^ -', '', raw)
|
|
if clean.find(' ') == -1 and clean.find('.') != -1:
|
|
clean = re.sub('\.', ' ', clean)
|
|
clean = re.sub('_', ' ', clean)
|
|
clean = re.sub('([\[\(_]|- )$', '', clean).strip()
|
|
|
|
self._part('title', [], raw, clean)
|
|
|
|
# Start process for end
|
|
clean = re.sub('(^[-\. ()]+)|([-\. ]+$)', '', self.excess_raw)
|
|
clean = re.sub('[\(\)\/]', ' ', clean)
|
|
match = re.split('\.\.+| +', clean)
|
|
if len(match) > 0 and isinstance(match[0], tuple):
|
|
match = list(match[0])
|
|
|
|
clean = filter(bool, match)
|
|
clean = [item for item in filter(lambda a: a != '-', clean)]
|
|
clean = [item.strip('-') for item in clean]
|
|
if len(clean) != 0:
|
|
group_pattern = clean[-1] + self.group_raw
|
|
if self.torrent['name'].find(group_pattern) == \
|
|
len(self.torrent['name']) - len(group_pattern):
|
|
self._late('group', clean.pop() + self.group_raw)
|
|
|
|
if 'map' in self.torrent.keys() and len(clean) != 0:
|
|
episode_name_pattern = (
|
|
'{episode}'
|
|
'' + re.sub('_+$', '', clean[0])
|
|
)
|
|
if self.torrent['map'].find(episode_name_pattern) != -1:
|
|
self._late('episodeName', clean.pop(0))
|
|
|
|
if len(clean) != 0:
|
|
if len(clean) == 1:
|
|
clean = clean[0]
|
|
self._part('excess', [], self.excess_raw, clean)
|
|
return self.parts
|