026

Author: pybites

026/.gitignore

@@ -0,0 +1 @@
+cache*

026/README.md

@@ -0,0 +1,49 @@
+## 026
+
+Interactive script to query the [OMDb API](http://www.omdbapi.com):
+
+	$ python omdb.py
+	Script to query OMDb API
+	Enter IMDB ID or title (q to exit): tt0104348
+
+	Title: Glengarry Glen Ross
+	Year: 1992
+	Genre: Crime, Drama, Mystery
+	Director: James Foley
+	Actors: Al Pacino, Jack Lemmon, Alec Baldwin, Alan Arkin
+	IMDB score: 7.8
+	Plot: An examination of the machinations behind the scenes at a real estate office.
+
+
+	Enter IMDB ID or title (q to exit): scarface
+	Year of release? 1983
+
+	Title: Scarface
+	Year: 1983
+	Genre: Crime, Drama
+	Director: Brian De Palma
+	Actors: Al Pacino, Steven Bauer, Michelle Pfeiffer, Mary Elizabeth Mastrantonio
+	IMDB score: 8.3
+	Plot: In Miami in 1980, a determined Cuban immigrant takes over a drug cartel and succumbs to greed.
+
+
+	Enter IMDB ID or title (q to exit): nonsense
+	Year of release? 111
+	Error: Movie not found!
+	Enter IMDB ID or title (q to exit): q
+	Bye
+	$ python omdb.py -h
+	Script to query OMDb API
+	Enter IMDB ID or title (q to exit): scarface
+	Year of release? 1983
+
+	<li>
+		<h1>Scarface (1983)</h1>
+		<h2>Genre: Crime, Drama / Director: Brian De Palma</h2>
+		<h3>Actors: Al Pacino, Steven Bauer, Michelle Pfeiffer, Mary Elizabeth Mastrantonio</h3>
+		<h4>IMDB score: 8.3</h4>
+		<p>In Miami in 1980, a determined Cuban immigrant takes over a drug cartel and succumbs to greed.</p>
+	</li>
+
+	Enter IMDB ID or title (q to exit): q
+	Bye

026/omdb.py

@@ -0,0 +1,54 @@
+from pprint import pprint as pp
+import sys
+
+import requests
+import requests_cache
+
+from templates import TEMPLATE, HTML_TEMPLATE
+
+OMDB = 'http://www.omdbapi.com'
+OMDB_BY_IMDB = OMDB + '/?i={imdb}'
+OMDB_BY_TITLE_YEAR = OMDB + '/?t={title}&y={year}'
+
+# cache repeated calls to OMDB (not sure about API limits)
+requests_cache.install_cache()
+
+
+def query_omdb(**kwargs):
+    if 'imdb' in kwargs:
+        url = OMDB_BY_IMDB.format(**kwargs)
+    else:
+        url = OMDB_BY_TITLE_YEAR.format(**kwargs)
+    return requests.get(url).json()
+
+
+if __name__ == '__main__':
+    args = sys.argv[1:]
+    html = True if '-h' in args else False
+    verbose = True if '-v' in args else False
+    answer = None
+
+    print('Script to query OMDb API')
+
+    while answer != 'q':
+        answer = input('Enter IMDB ID or title (q to exit): ').lower()
+        params = dict()
+
+        if answer == 'q':
+            print('Bye')
+            break
+        elif 'tt' in answer:
+            params['imdb'] = answer
+        else:
+            params['title'] = answer
+            params['year'] = input('Year of release? ')
+
+        resp = query_omdb(**params)
+        if verbose: 
+            pp(resp)
+
+        if 'Error' in resp:
+            print('Error: {}'.format(resp['Error']))
+        else:
+            tmpl = HTML_TEMPLATE if html else TEMPLATE
+            print(tmpl.format(**dict(resp)))

026/requirements.txt

@@ -0,0 +1,2 @@
+requests==2.13.0
+requests-cache==0.4.13

026/templates.py

@@ -0,0 +1,22 @@
+import os
+import sys
+
+TEMPLATE = '''
+Title: {Title}
+Year: {Year}
+Genre: {Genre}
+Director: {Director}
+Actors: {Actors}
+IMDB score: {imdbRating}
+Plot: {Plot}
+
+'''
+HTML_TEMPLATE = '''
+<li>
+    <h1>{Title} ({Year})</h1>
+    <h2>Genre: {Genre} / Director: {Director}</h2>
+    <h3>Actors: {Actors}</h3>
+    <h4>IMDB score: {imdbRating}</h4>
+    <p>{Plot}</p>
+</li>
+'''

LOG.md

@@ -27,7 +27,7 @@
 | 023 | Apr 21, 2017 | [use Counter to count the most common words in a file](023) | we did some time ago, collections module is awesome |
 | 024 | Apr 22, 2017 | [generate color hex codes from random RGBs and color terminal text](024) | nice play with generators, RGB to hex with format, and colorizing the terminal with the [colored](https://pypi.python.org/pypi/colored) package. This could be useful for future cli apps | 
 | 025 | Apr 23, 2017 | [Simple test #database generator script #python #sqlite #contextmanager](025) | Script to generate a quick, test sqlite3 database with 1 table and 3 columns. Can be customised and expanded to suit your needs. Definitely useful for playing around with persistent data. |
-| 026 | Apr 24, 2017 | [TITLE](026) | LEARNING |
+| 026 | Apr 24, 2017 | [Simple script to retrieve #movie data from OMDb #API](026) | Get OMDB for movie data, query API by IMDB ID or title. Returns text or html. Uses request_cache to limit repeated calls to API. Could help some folks taking this [this week's challenge](http://pybit.es/codechallenge16.html) |
 | 027 | Apr 25, 2017 | [TITLE](027) | LEARNING |
 | 028 | Apr 26, 2017 | [TITLE](028) | LEARNING |
 | 029 | Apr 27, 2017 | [TITLE](029) | LEARNING |