Fixes #22 - Add static files to setup.py

This fixes issues involved with deployment via bdist_egg.
This commit is contained in:
Anthony Miyaguchi 2017-05-03 18:36:07 +00:00 коммит произвёл Ryan Harter
Родитель 5f5c7c23ca
Коммит ebc120ed68
3 изменённых файлов: 8 добавлений и 9 удалений

1
MANIFEST.in Normal file
Просмотреть файл

@ -0,0 +1 @@
recursive-include mozetl *.json

Просмотреть файл

@ -1,9 +1,11 @@
import json
import os
from pyspark.sql.types import StructType
import pkg_resources
import mozetl.topline
def schema_from_json(path, relative=True):
def schema_from_json(path):
""" Create a pyspark schema from the json representation.
The json representation must be from a StructType. This can be
@ -16,14 +18,9 @@ def schema_from_json(path, relative=True):
>>> json_data = spark.read.parquet(path).schema.json()
:path str: Path the the json data
:relative bool: Use the relative path to the current file.
"""
if relative:
path = os.path.join(os.path.dirname(__file__), path)
with open(path) as json_data:
data = json.load(json_data)
json_data = pkg_resources.resource_stream(mozetl.topline.__name__, path)
data = json.load(json_data)
return StructType.fromJson(data)

Просмотреть файл

@ -10,6 +10,7 @@ setup(
author_email='harterrt@mozilla.com',
url='https://github.com/mozilla/python_mozetl.git',
packages=find_packages(exclude=['tests']),
include_package_data=True,
install_requires=[
'python_moztelemetry', # TODO: pin version
'click',