Properly handle extra source files

pirate/data.py

@@ -1,7 +1,13 @@
 import json
+import pkgutil
 
-categories = json.load(open('data/categories.json'))
-sorts = json.load(open('data/sorts.json'))
+
+def get_resource(filename):
+    return pkgutil.get_data(__package__, 'data/' + filename)
+
+
+categories = json.loads(get_resource('categories.json').decode())
+sorts = json.loads(get_resource('sorts.json').decode())
 
 default_headers = {'User-Agent': 'pirate get'}
 default_timeout = 10

setup.py

@@ -9,6 +9,7 @@ setup(name='pirate-get',
     author_email='me@viktorstanchev.com',
     license='GPL',
     packages=find_packages(),
+    package_data={'': ["data/*.json"]},
     entry_points={
         'console_scripts': ['pirate-get = pirate.pirate:main']
     },