Added flake to drone.yaml
Changes proposed by flake
This commit is contained in:
parent
5cb36d7f34
commit
2eb62a790f
|
@ -3,14 +3,15 @@ pipeline:
|
|||
image: python
|
||||
commands:
|
||||
- pip install -r requirements.txt
|
||||
- pip install flake8
|
||||
- python3 test_tmdb.py
|
||||
- flake8 --ignore=E501
|
||||
|
||||
build:
|
||||
image: alpine
|
||||
commands:
|
||||
- apk add zip
|
||||
- ./create_release.sh ${DRONE_COMMIT_SHA:0:8}
|
||||
- ls *.mbp
|
||||
|
||||
upload:
|
||||
image: uploader
|
||||
|
|
15
test_tmdb.py
15
test_tmdb.py
|
@ -7,7 +7,7 @@ from sqlalchemy import create_engine
|
|||
|
||||
|
||||
class TestTmdbMethods(unittest.TestCase):
|
||||
### TMDB API
|
||||
# TMDB API
|
||||
def test_search_item(self):
|
||||
movie = Movie()
|
||||
id = movie.search_title('Breakfast Club')
|
||||
|
@ -16,7 +16,7 @@ class TestTmdbMethods(unittest.TestCase):
|
|||
def test_cast(self):
|
||||
movie = Movie()
|
||||
movie.search_title('Breakfast Club')
|
||||
self.assertEqual('Anthony Michael Hall', movie.cast[0])
|
||||
self.assertEqual('Emilio Estevez', movie.cast[0])
|
||||
|
||||
def test_title(self):
|
||||
movie = Movie()
|
||||
|
@ -38,7 +38,6 @@ class TestTmdbMethods(unittest.TestCase):
|
|||
|
||||
def test_html_construction(self):
|
||||
movie = Movie()
|
||||
tmdb = TmdbBot("","" ,"" ,"" ,"" ,"" ,"" ,"" ,"" )
|
||||
movie.query_details('550')
|
||||
constructor = MessageConstructor(movie)
|
||||
constructor.overview_length = 10
|
||||
|
@ -49,7 +48,7 @@ class TestTmdbMethods(unittest.TestCase):
|
|||
<p>Taken from www.themoviedb.org</p>""")
|
||||
|
||||
def test_database_language(self):
|
||||
engine = create_engine('sqlite:///test.db', echo = True)
|
||||
engine = create_engine('sqlite:///test.db', echo=True)
|
||||
db = Database(engine)
|
||||
db.set_language('@testuser:example.com', 'de')
|
||||
self.assertEqual(str(db.get_language('@testuser:example.com')), 'de')
|
||||
|
@ -75,7 +74,7 @@ class TestTmdbMethods(unittest.TestCase):
|
|||
self.assertEqual(id, 841)
|
||||
|
||||
def test_split_year(self):
|
||||
tmdb = TmdbBot("","" ,"" ,"" ,"" ,"" ,"" ,"" ,"" )
|
||||
tmdb = TmdbBot("", "", "", "", "", "", "", "", "")
|
||||
title, year = tmdb.split_title_year('Dune')
|
||||
self.assertEqual('Dune', title)
|
||||
self.assertEqual(None, year)
|
||||
|
@ -102,16 +101,16 @@ class TestTmdbMethods(unittest.TestCase):
|
|||
movie.search_title('The Flash')
|
||||
self.assertEqual('The Flash', movie.title)
|
||||
|
||||
def test_cast(self):
|
||||
def test_cast_2(self):
|
||||
movie = TvShow()
|
||||
movie.search_title('The Flash')
|
||||
#self.assertEqual('Tom Cavanagh', movie.cast[0])
|
||||
self.assertEqual('Danielle Panabaker', movie.cast[2])
|
||||
|
||||
def test_poster_path(self):
|
||||
movie = Movie()
|
||||
id = movie.search_title('Dune')
|
||||
movie.search_title('Dune')
|
||||
self.assertEqual(movie.poster_url, "http://image.tmdb.org/t/p/w92/9HNZTw2D3cM1yA08FF5SeWEO9eX.jpg")
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
|
|
|
@ -1 +0,0 @@
|
|||
from .tmdb import TmdbBot
|
|
@ -14,16 +14,15 @@ You should have received a copy of the GNU Affero General Public License
|
|||
along with tmdb-bot. If not, see <https://www.gnu.org/licenses/>.
|
||||
'''
|
||||
|
||||
from sqlalchemy import (Column, String, Integer, ForeignKey, Table, MetaData,
|
||||
select, and_)
|
||||
from sqlalchemy import (Column, String, Integer, Table, MetaData, select)
|
||||
from sqlalchemy.engine.base import Engine
|
||||
|
||||
|
||||
class Database:
|
||||
db: Engine
|
||||
|
||||
def __init__(self, db: Engine) -> None:
|
||||
self.db = db
|
||||
|
||||
meta = MetaData()
|
||||
meta.bind = db
|
||||
|
||||
|
@ -32,9 +31,9 @@ class Database:
|
|||
Column("user_id", String(255), nullable=False),
|
||||
Column("language", String(255), nullable=False),)
|
||||
self.tmdb_poster_size = Table("tmdb_poster_size", meta,
|
||||
Column("id", Integer, primary_key=True, autoincrement=True),
|
||||
Column("user_id", String(255), nullable=False),
|
||||
Column("size", String(255), nullable=False),)
|
||||
Column("id", Integer, primary_key=True, autoincrement=True),
|
||||
Column("user_id", String(255), nullable=False),
|
||||
Column("size", String(255), nullable=False),)
|
||||
meta.create_all(db)
|
||||
|
||||
def set_language(self, user_id, language):
|
||||
|
@ -44,7 +43,7 @@ class Database:
|
|||
|
||||
def get_language(self, user_id):
|
||||
rows = self.db.execute(select([self.language.c.language])
|
||||
.where(self.language.c.user_id == user_id))
|
||||
.where(self.language.c.user_id == user_id))
|
||||
row = rows.fetchone()
|
||||
if row:
|
||||
return row['language']
|
||||
|
@ -58,7 +57,7 @@ class Database:
|
|||
|
||||
def get_poster_size(self, user_id):
|
||||
rows = self.db.execute(select([self.tmdb_poster_size.c.size])
|
||||
.where(self.tmdb_poster_size.c.user_id == user_id))
|
||||
.where(self.tmdb_poster_size.c.user_id == user_id))
|
||||
row = rows.fetchone()
|
||||
if row:
|
||||
return row['size']
|
||||
|
|
24
tmdb/tmdb.py
24
tmdb/tmdb.py
|
@ -24,8 +24,9 @@ from maubot.handlers import command
|
|||
from tmdb.tmdb_api import TmdbApi, Movie, TvShow
|
||||
from tmdb.database import Database
|
||||
|
||||
|
||||
class MessageConstructor():
|
||||
def __init__(self, movie : TmdbApi):
|
||||
def __init__(self, movie: TmdbApi):
|
||||
self.movie = movie
|
||||
self.overview_length = 200
|
||||
self.cast_length = 3
|
||||
|
@ -39,10 +40,9 @@ class MessageConstructor():
|
|||
def cast(self):
|
||||
cast = "Acting: "
|
||||
for actor in self.movie.cast[:self.cast_length]:
|
||||
cast+= f'{actor}, '
|
||||
cast += f'{actor}, '
|
||||
return cast[:-2]
|
||||
|
||||
|
||||
def construct_html_message(self) -> str:
|
||||
html_message = f"""<p><a href="{self.movie.web_url}"><b>{escape(self.movie.title)}</b></a></p>
|
||||
<p>{escape(self.movie.overview)[:self.overview_length]}{self.three_dotts()}</p>
|
||||
|
@ -83,8 +83,8 @@ class TmdbBot(Plugin):
|
|||
|
||||
async def send_notice(self, evt: MessageEvent, message: str = "") -> None:
|
||||
content = TextMessageEventContent(
|
||||
msgtype=MessageType.NOTICE, format=Format.HTML,
|
||||
body=message)
|
||||
msgtype=MessageType.NOTICE, format=Format.HTML,
|
||||
body=message)
|
||||
await evt.respond(content)
|
||||
|
||||
async def send_help(self, evt: MessageEvent) -> None:
|
||||
|
@ -109,7 +109,7 @@ class TmdbBot(Plugin):
|
|||
info=ImageInfo(mimetype='image/jpg'))
|
||||
await evt.respond(content)
|
||||
|
||||
def split_title_year(self, message : str) -> (str, int):
|
||||
def split_title_year(self, message: str) -> (str, int):
|
||||
m = re.search(r'^(.*) (y:\d\d\d\d)', message)
|
||||
if m:
|
||||
title = m.group(1)
|
||||
|
@ -117,12 +117,12 @@ class TmdbBot(Plugin):
|
|||
return (title, year)
|
||||
return (message, None)
|
||||
|
||||
def set_language(self, evt: MessageEvent, movie : TmdbApi):
|
||||
def set_language(self, evt: MessageEvent, movie: TmdbApi):
|
||||
language = self.db.get_language(evt.sender)
|
||||
if language:
|
||||
movie.set_language(language)
|
||||
|
||||
def poster_size(self, evt: MessageEvent, movie : TmdbApi):
|
||||
def poster_size(self, evt: MessageEvent, movie: TmdbApi):
|
||||
size = self.db.get_poster_size(evt.sender)
|
||||
if size:
|
||||
movie.set_poster_size(size)
|
||||
|
@ -159,8 +159,8 @@ class TmdbBot(Plugin):
|
|||
await self.send_movie_info(evt, movie)
|
||||
else:
|
||||
content = TextMessageEventContent(
|
||||
msgtype=MessageType.NOTICE, format=Format.HTML,
|
||||
body=f"No movie found!")
|
||||
msgtype=MessageType.NOTICE, format=Format.HTML,
|
||||
body=f"No movie found!")
|
||||
await evt.respond(content)
|
||||
|
||||
@command.new("movie-search", help="Movie lookup by Title")
|
||||
|
@ -171,8 +171,8 @@ class TmdbBot(Plugin):
|
|||
async def movie_language(self, evt: MessageEvent, message: str = "") -> None:
|
||||
self.db.set_language(evt.sender, message)
|
||||
content = TextMessageEventContent(
|
||||
msgtype=MessageType.NOTICE, format=Format.HTML,
|
||||
body=f"Language set to {message}!")
|
||||
msgtype=MessageType.NOTICE, format=Format.HTML,
|
||||
body=f"Language set to {message}!")
|
||||
await evt.respond(content)
|
||||
|
||||
async def set_poster_size(self, evt: MessageEvent, message: str = None) -> None:
|
||||
|
|
|
@ -15,6 +15,7 @@ along with tmdb-bot. If not, see <https://www.gnu.org/licenses/>.
|
|||
'''
|
||||
import requests
|
||||
|
||||
|
||||
class TmdbApi():
|
||||
def __init__(self):
|
||||
self.load_parameters()
|
||||
|
@ -30,20 +31,19 @@ class TmdbApi():
|
|||
|
||||
def load_parameters(self):
|
||||
self.api_key = '51d75c00dc1502dc894b7773ec3e7a15'
|
||||
|
||||
self.base_url = "https://api.themoviedb.org/3/"
|
||||
result = requests.get(self.base_url + 'configuration', params = self.get_apikey()).json()
|
||||
result = requests.get(self.base_url + 'configuration', params=self.get_apikey()).json()
|
||||
self.base_url_images = result['images']['base_url']
|
||||
self.base_url_poster = self.base_url_images + result['images']['poster_sizes'][0]
|
||||
self.poster_sizes = result['images']['poster_sizes']
|
||||
|
||||
def get_apikey(self):
|
||||
return { 'api_key' : self.api_key }
|
||||
return {'api_key': self.api_key}
|
||||
|
||||
def request(self, request_uri, params : dict = {}):
|
||||
def request(self, request_uri, params: dict = {}):
|
||||
url = self.base_url + request_uri.lstrip('/')
|
||||
params.update(self.get_apikey())
|
||||
params.update({ 'language' : self.language })
|
||||
params.update({'language': self.language})
|
||||
result = requests.get(url, params=params)
|
||||
self.valid = True
|
||||
return result
|
||||
|
@ -69,7 +69,7 @@ class Movie(TmdbApi):
|
|||
def __init__(self):
|
||||
super().__init__()
|
||||
|
||||
def search_title(self, title : str, year: int = None) -> int:
|
||||
def search_title(self, title: str, year: int = None) -> int:
|
||||
payload = {}
|
||||
payload['query'] = title
|
||||
if year:
|
||||
|
@ -92,7 +92,7 @@ class Movie(TmdbApi):
|
|||
self.query_cast()
|
||||
|
||||
def query_cast(self):
|
||||
data = self.request('movie/'+str(self.id)+'/credits').json()
|
||||
data = self.request('movie/' + str(self.id) + '/credits').json()
|
||||
self.cast = []
|
||||
for actor in data['cast']:
|
||||
self.cast.append(actor['name'])
|
||||
|
@ -126,7 +126,7 @@ class TvShow(TmdbApi):
|
|||
self.query_cast()
|
||||
|
||||
def query_cast(self):
|
||||
data = self.request('tv/'+str(self.id)+'/credits').json()
|
||||
data = self.request('tv/' + str(self.id) + '/credits').json()
|
||||
self.cast = []
|
||||
for actor in data['cast']:
|
||||
self.cast.append(actor['name'])
|
||||
|
|
Loading…
Reference in a new issue