Scott Faria
8 years ago
committed by
GitHub
GitHub
2 changed files with 184 additions and 0 deletions
@ -0,0 +1,176 @@ |
|||
import re |
|||
from bs4 import BeautifulSoup |
|||
from couchpotato.core.media._base.providers.torrent.base import TorrentProvider |
|||
from couchpotato.core.helpers.variable import tryInt |
|||
from couchpotato.core.helpers.encoding import tryUrlencode |
|||
from couchpotato.core.logger import CPLog |
|||
|
|||
__author__ = 'Scott Faria - scott.faria@gmail.com' |
|||
|
|||
log = CPLog(__name__) |
|||
|
|||
|
|||
class Base(TorrentProvider): |
|||
|
|||
http_time_between_calls = 1 |
|||
|
|||
urls = { |
|||
'test': 'http://goem.org/', |
|||
'base_url': 'http://goem.org', |
|||
'login': 'http://goem.org/takelogin.php', |
|||
'login_check': 'http://goem.org/my.php', |
|||
'search': 'http://goem.org/advanced.php?action=search&imdb=%s&title=&title-tag=%s&title-tag-type=any&year=%d&page=%d', |
|||
} |
|||
|
|||
source = '&source%%5B%%5D=%s' |
|||
|
|||
def _find_quality_params(self, quality_id): |
|||
quality_id = quality_id.upper() |
|||
if quality_id == 'HD': |
|||
return {'tag': '1080P;720P', 'param': False} |
|||
elif quality_id == 'SD': |
|||
return {'tag': '-1080P;-720P', 'param': False} |
|||
elif quality_id == '1080P': |
|||
return {'tag': '1080P', 'param': False} |
|||
elif quality_id == '720P': |
|||
return {'tag': '720P', 'param': False} |
|||
elif quality_id in ['DVDR', 'DVDRIP']: |
|||
return {'tag': 'DVD', 'param': True} |
|||
elif quality_id in ['BRDISK', 'BRRIP']: |
|||
return {'tag': 'BluRay', 'param': True} |
|||
elif quality_id in ['SCREENER', 'R5', 'TELECINE', 'TELESYNC', 'CAM']: |
|||
return None # Not allowed to be uploaded to goem |
|||
else: |
|||
return None |
|||
|
|||
def _get_page_count(self, html): |
|||
nav_div = html.find('p', attrs={'class', 'pager'}) |
|||
if nav_div: |
|||
page_links = nav_div.find_all('a') |
|||
if page_links: |
|||
pages = [0] |
|||
for link in page_links: |
|||
matcher = re.search('page=(\d*)', link['href']) |
|||
pages.append(tryInt(matcher.group(0))) |
|||
return max(pages) |
|||
return 1 |
|||
|
|||
def _add_torrent(self, table_row, results): |
|||
torrent_tds = table_row.find_all('td') |
|||
if len(torrent_tds) == 11: |
|||
url_base = self.urls['base_url'] |
|||
torrent_id = torrent_tds[0].find('a')['href'].replace('/details.php?id=', '') |
|||
download_url = url_base + torrent_tds[1].find('a')['href'] |
|||
details_link = torrent_tds[5].find('a') |
|||
details_url = url_base + details_link['href'] |
|||
torrent_name = details_link.string |
|||
torrent_seeders = tryInt(re.sub('[^0-9]', '', unicode(torrent_tds[7].find('nobr').contents[0].string))) |
|||
torrent_leechers = tryInt(re.sub('[^0-9]', '', unicode(torrent_tds[7].find('nobr').contents[2].string))) |
|||
torrent_size = self.parseSize(torrent_tds[9].find('span').string) |
|||
|
|||
results.append({ |
|||
'id': torrent_id, |
|||
'name': torrent_name, |
|||
'url': download_url, |
|||
'detail_url': details_url, |
|||
'size': torrent_size, |
|||
'seeders': torrent_seeders, |
|||
'leechers': torrent_leechers, |
|||
}) |
|||
|
|||
def _format_url(self, current_page, imdb_id, year, quality_tag, use_source_tag): |
|||
if use_source_tag: |
|||
return self.urls['search'] % (imdb_id, "", year, current_page) + self.source % quality_tag |
|||
else: |
|||
return self.urls['search'] % (imdb_id, quality_tag, year, current_page) |
|||
|
|||
# noinspection PyBroadException |
|||
def _search(self, movie, quality, results): |
|||
quality_map = self._find_quality_params(quality['identifier']) |
|||
if quality_map is None: |
|||
return |
|||
|
|||
quality = quality_map['tag'] |
|||
use_source_tag = quality_map['param'] |
|||
|
|||
year = movie['library']['year'] |
|||
imdb_id = movie['library']['identifier'] |
|||
|
|||
current_page = 1 |
|||
pages = -1 |
|||
|
|||
while True: |
|||
try: |
|||
url = self._format_url(current_page, imdb_id, year, quality, use_source_tag) |
|||
data = self.getHTMLData(url, opener=self.login_opener) |
|||
|
|||
if data: |
|||
html = BeautifulSoup(data) |
|||
if pages == -1: |
|||
pages = self._get_page_count(html) |
|||
|
|||
torrent_table = html.find('table', attrs={'id': 'browse'}) |
|||
if torrent_table: |
|||
torrent_rows = torrent_table.find_all('tr', attrs={'class': 'table_row'}) |
|||
for row in torrent_rows: |
|||
self._add_torrent(row, results) |
|||
|
|||
if current_page >= pages: |
|||
break |
|||
except: |
|||
log.error('Unexpected error while searching %s: %s', (self.getName(), traceback.format_exc())) |
|||
return |
|||
|
|||
def getLoginParams(self): |
|||
return tryUrlencode({ |
|||
'goeser': self.conf('username'), |
|||
'gassmord': self.conf('password'), |
|||
'login': 'submit', |
|||
}) |
|||
|
|||
def loginSuccess(self, output): |
|||
return 'Login failed!' not in output.lower() |
|||
|
|||
def isEnabled(self): |
|||
return super(Base, self).isEnabled() and self.getDomain() |
|||
|
|||
|
|||
config = [{ |
|||
'name': 'goem', |
|||
'groups': [ |
|||
{ |
|||
'tab': 'searcher', |
|||
'list': 'torrent_providers', |
|||
'name': 'Goem', |
|||
'description': 'See <a href="http://www.goem.org">Goem</a>', |
|||
'icon': 'AAABAAEAEBAAAAEAIABoBAAAFgAAACgAAAAQAAAAIAAAAAEAIAAAAAAAAAQAAAAAAAAAAAAAAAAA AAAAAAD///8A////AOrp6BXm5uQZ////AP///wD///8A////AP///wD///8A////AP///wD///8A ////AP///wD///8A/Pz8AdTW0V6SfmHwl4px/////wD///8A////AP///wD///8A////AP///wD/ //8A////AP///wD///8A////AODd1iibgFn/0quC/59+Xv/Z3Ngm1tXOPqGZiud9dWH9cmpX/YSA c/25uLFi9PT2Cv///wD///8A////AP///wD08/IMsZ12//bruv/Epnv/c2dO/3ZpSf2QhGP93M+q +v3yxvj36r34xrmP+pOIb/3ExL1k////AP///wD///8A////AMe+s2qok3b/nXdc/2hRN/3WzrL4 +vLQ9c3FuvWYhmv47OS89f7+1/X+/tH3xLeQ+r68svf///8A////AP///wD///8Ap4xq/cqrff3L uaP4/v7y88zBqvX1+f5il496uZuNcff068rwt6eW98Kzj/jEt5XQ2NnWKP///wD///8A////AIx6 YP2sm3z6/v788/7++e3Sx7Pub2BG931fPfiaiG/ztauS8Nvd3F6Lfl6269+5+KyjkfX6+voD//// APDw8A6il3X9rJ6K+vLr2/OQgGf30siz7cq/q+qllX3u/vni48Cylu5jVj7ucGRE6eXbu/fHup36 7OvqFP///wDp6OYYsqWD/c7Ct/i5oYP3g2VLyYRmRuX38eLg/v763f7+9t3+/vXjtaeQ8LWtk/X+ /vL149m6+t7e2Kv///8A9PT0Cryvjv3TysD3nIx5+H5gRO1bQiL65NvL5ff0495yYXSj6+TQcP7+ 7umtn433r6KG+O/lyvrc2dTn////AP///wDAtZv90caz+P7+/vOmlnz3oJOF8/7+/uP+/vnZgHB2 ofPu3+nKv6vzr66jkYx7XK3Pwar66enkGf///wD///8AzcO4Tca6nfj+/v71/v7+8LSfjvWNe2b3 9/Tu6f7+/un+/v7q7Oba8F9TPPWDclX10cSz/fb29gf///8A////APz9/AHEtp3908m7+vv4+fWW b0r6qZNnu416Z+DZ0cfzbVI7+qyehOr++/L3/Pbr+tTPxf3///8A////AP///wD///8A7OnoFb2t kv3m39f6jH9t/VpFKf21opH4q5+M+JiRgKiThmat/v77+sm+sP38/P4B////AP///wD///8A//// AP///wD29vYHw7Wj/cGxof3t5+L6/v7++Pj39fihlIT9va+e/dbMw/36+/wD////AP///wD///8A ////AP///wD///8A////AP///wDy8O4P2dPL/dnTy/3d19D94NjR/fDt7BL///8A////AP///wD/ //8A//8AAM//AACMPwAAgA8AAMADAADBAwAAwBEAAMABAADAAAAAwCAAAMABAADgAQAA4AEAAPAD AAD4BwAA/h8AAA==', |
|||
'wizard': True, |
|||
'options': [ |
|||
{ |
|||
'name': 'enabled', |
|||
'type': 'enabler', |
|||
'default': False, |
|||
}, |
|||
{ |
|||
'name': 'username', |
|||
'default': '', |
|||
}, |
|||
{ |
|||
'name': 'password', |
|||
'default': '', |
|||
'type': 'password', |
|||
}, |
|||
{ |
|||
'name': 'extra_score', |
|||
'advanced': True, |
|||
'label': 'Extra Score', |
|||
'type': 'int', |
|||
'default': 0, |
|||
'description': 'Starting score for each release found via this provider.', |
|||
} |
|||
], |
|||
}, |
|||
], |
|||
}] |
|||
|
|||
|
|||
@ -0,0 +1,8 @@ |
|||
from couchpotato.core.media._base.providers.torrent.goem import Base |
|||
from couchpotato.core.media.movie.providers.base import MovieProvider |
|||
|
|||
autoload = 'Goem' |
|||
|
|||
|
|||
class Goem(MovieProvider, Base): |
|||
pass |
|||
Loading…
Reference in new issue