arti/KaruMediaServer
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge branch 'master' of https://git.wut.ee/arti/KaruMediaServer

Browse Source
This commit is contained in:
Arti Zirk 2016-11-14 11:23:04 +02:00
commit 66a5f10813
3 changed files with 26 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
.gitignore vendored
View File

@ -88,3 +88,6 @@ ENV/
# Rope project settings # Rope project settings
.ropeproject .ropeproject
static/Filmid/
static/filmid/

1
api/karumedia/controllers.py
View File

@ -48,6 +48,7 @@ class MoviesCollection(BaseResource):
"synopsis": metadata["plot_outline"], "synopsis": metadata["plot_outline"],
"mpa_rating": metadata["certification"], "mpa_rating": metadata["certification"],
"genres": metadata["genres"], "genres": metadata["genres"],
"yt_trailer_code": metadata["yt_trailer_code"],
"state": "ok" "state": "ok"
} }
try: try:

27
metainfo-dl.py
View File

@ -7,13 +7,17 @@ import requests
from pprint import pprint from pprint import pprint
import re import re
import logging import logging
from logging import info, debug, warning from logging import info, debug, warning, error
import subprocess import subprocess
logging.basicConfig(level=logging.DEBUG) logging.basicConfig(level=logging.DEBUG)
movie_dir = "/home/arti/Videod/Filmid" movie_dir = "/home/arti/Videod/Filmid"
YTKEY = os.environ.get("YTKEY")
if not YTKEY:
error("YTKEY not set")
exit(1)
imdb = Imdb() imdb = Imdb()
@ -42,7 +46,7 @@ def write_metadata(movie_dir, mm):
"genres": mm.genres, "genres": mm.genres,
"plots": mm.plots, "plots": mm.plots,
"imdb_id":str(mm.imdb_id)} "imdb_id":str(mm.imdb_id)}
info("opening metadata.json in {} for {}".format(movie_dir, metadata["title"])) info("opening metadata.json in {} for {}".format(movie_dir, metadata["title"]))
with open(movie_dir+"/metadata.json", "w") as f: with open(movie_dir+"/metadata.json", "w") as f:
info("writing metadata for {}".format(metadata["title"])) info("writing metadata for {}".format(metadata["title"]))
@ -76,14 +80,30 @@ def download_trailer(movie_dir, metadata):
subprocess.call(['wget', trailer[2], subprocess.call(['wget', trailer[2],
"-O", movie_dir+"/trailer.mp4"]) "-O", movie_dir+"/trailer.mp4"])
def add_yt_trailer_code(md):
params = {"key": YTKEY,
"part":"id", "maxResults":1,
"q":"{} ({}) trailer".format(md.get("title"), md.get("year", ""))}
r = requests.get("https://www.googleapis.com/youtube/v3/search", params=params)
md["yt_trailer_code"] = r.json()["items"][0]["id"]["videoId"]
return md
def metadata_update_needed(metadata_file): def metadata_update_needed(metadata_file):
with open(metadata_file, "r") as f: with open(metadata_file, "r") as f:
md = json.loads(f.read()) md = json.loads(f.read())
if "imdb_id" not in md: if "imdb_id" not in md:
return True return True
elif "yt_trailer_code" not in md:
md = add_yt_trailer_code(md)
print(md.get("title"), md.get("yt_trailer_code"))
fd = open(metadata_file, "w")
fd.write(json.dumps(md, indent=4, sort_keys=True))
fd.close()
else: else:
return False return False
for movie in os.listdir(movie_dir): for movie in os.listdir(movie_dir):
#print(movie) #print(movie)
if os.path.isfile(movie_dir+"/"+movie+"/metadata.json") \ if os.path.isfile(movie_dir+"/"+movie+"/metadata.json") \
@ -103,6 +123,3 @@ for movie in os.listdir(movie_dir):
except: except:
logging.exception("Poster download failed") logging.exception("Poster download failed")
#download_trailer(movie_dir+"/"+movie, mm) #download_trailer(movie_dir+"/"+movie, mm)