Refactor include path.

1 files changed, 6 insertions, 1 deletions

diff --git a/setup.py b/setup.py
index e5651a0..65ca412 100644
--- a/setup.py
+++ b/setup.py
@@ -1,8 +1,13 @@
 from distutils.core import setup, Extension
+import os
 
 version = '0.0.1'
 
-msgpack_mod = Extension('msgpack', sources=['msgpack.c'])
+PACKAGE_ROOT = os.getcwdu()
+INCLUDE_PATH = os.path.join(PACKAGE_ROOT, 'include')
+msgpack_mod = Extension('msgpack',
+                        sources=['msgpack.c'],
+                        include_dirs=[INCLUDE_PATH])
 
 desc = 'MessagePack serializer/desirializer.'
 long_desc = desc + """