fixed issue where settings.yaml cannot be detected

hiimivantang · Aug 29, 2016 · 996e2f9 · 996e2f9
1 parent 489fb52
commit 996e2f9
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 2 deletions.
diff --git a/ltadatamallcrawler/config.py b/ltadatamallcrawler/config.py
@@ -26,7 +26,7 @@ def eprint(*args, **kwargs):
     print(*args, file=sys.stderr, **kwargs)
 
 
-with open('settings.yaml', 'r') as stream:
+with open(os.path.dirname(__file__) + '/' + 'settings.yaml', 'r') as stream:
     try:
         settings = yaml.load(stream)
     except yaml.YAMLError as e:

diff --git a/requirements-in.txt b/requirements-in.txt
@@ -1 +1,2 @@
 requests
+pyyaml
diff --git a/requirements.txt b/requirements.txt
@@ -1 +1,2 @@
 requests
+pyyaml
diff --git a/setup.py b/setup.py
@@ -35,9 +35,12 @@
     author_email='hiimivantang@gmail.com',
     url='https://github.com/hiimivantang/ltadatamallcrawler',
     packages=find_packages(exclude=['contrib', 'docs', 'tests']),
+    #package_data={
+    #  'ltadatamallcrawler':['urls.csv','settings.yaml']
+    #},
     entry_points={
         'console_scripts':[
-            'ltadatamallcrawler=ltadatamallcrawler.cli:cli',
+            'lta-datamall-crawler=ltadatamallcrawler.requestor:main',
             ],
         },
     include_package_data=True,