From cde62c5db394f752d2cd4c23fd1bba30b3bcb1bc Mon Sep 17 00:00:00 2001
From: Zied Aouini <aouinizied@gmail.com>
Date: Fri, 20 Dec 2019 16:15:27 +0100
Subject: [PATCH] Initial Windows build.

---
 setup.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 01932b8..bbc87dd 100644
--- a/setup.py
+++ b/setup.py
@@ -90,6 +90,8 @@ try:
 except ImportError:
     bdist_wheel = None
 
+cmdclass = {'nDPI': BuildNdpiCommand, 'build_py': BuildPyCommand, 'bdist_wheel': bdist_wheel} if bdist_wheel is not None else dict()
+
 setup(
     name="nfstream",
     version='3.0.4',
@@ -102,7 +104,7 @@ setup(
     author_email='aouinizied@gmail.com',
     packages=['nfstream'],
     install_requires=install_requires,
-    cmdclass={'nDPI': BuildNdpiCommand, 'build_py': BuildPyCommand, 'bdist_wheel': bdist_wheel},
+    cmdclass=cmdclass,
     setup_requires=pytest_runner,
     tests_require=['pytest>=5.0.1'],
     include_package_data=True,