-
Notifications
You must be signed in to change notification settings - Fork 903
/
setup.py
executable file
·28 lines (23 loc) · 1005 Bytes
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
#!/usr/bin/env python
import os
from setuptools import setup
from distutils.core import Extension
import platform
work_dir = os.path.dirname(os.path.realpath(__file__))
mod_dir = os.path.join(work_dir, 'src', 'confluent_kafka')
ext_dir = os.path.join(mod_dir, 'src')
# On Un*x the library is linked as -lrdkafka,
# while on windows we need the full librdkafka name.
if platform.system() == 'Windows':
librdkafka_libname = 'librdkafka'
else:
librdkafka_libname = 'rdkafka'
module = Extension('confluent_kafka.cimpl',
libraries=[librdkafka_libname],
sources=[os.path.join(ext_dir, 'confluent_kafka.c'),
os.path.join(ext_dir, 'Producer.c'),
os.path.join(ext_dir, 'Consumer.c'),
os.path.join(ext_dir, 'Metadata.c'),
os.path.join(ext_dir, 'AdminTypes.c'),
os.path.join(ext_dir, 'Admin.c')])
setup(ext_modules=[module])