hiredis package option

haliphax · Nov 3, 2016 · f634bab · f634bab
1 parent 031e815
commit f634bab
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/rodario/hiredis_requirements.txt b/rodario/hiredis_requirements.txt
@@ -0,0 +1 @@
+hiredis==0.2.0
diff --git a/rodario/setup.py b/rodario/setup.py
@@ -6,14 +6,17 @@
 
 if __name__ == '__main__':
     reqs = []
-    test_reqs = []
+    extras = {}
     abspath = realpath(dirname(__file__))
 
     with open(join(abspath, 'requirements.txt')) as reqfile:
         reqs = reqfile.readlines()
 
-    with open(join(abspath, 'test_requirements.txt')) as reqfile:
-        test_reqs = reqfile.readlines()
+    for extra in ('test', 'hiredis',):
+        filename = '{extra}_requirements.txt'.format(extra=extra)
+
+        with open(join(abspath, filename)) as reqfile:
+            extras[extra] = reqfile.readlines()
 
     setup(
         name='rodario',
@@ -33,7 +36,5 @@
         keywords='actor framework',
         packages=find_packages(),
         install_requires=reqs,
-        extras_require={
-            'test': test_reqs,
-        }
+        extras_require=extras
     )