diff --git a/setup.py b/setup.py index 818af1a2..1ae7d6b7 100755 --- a/setup.py +++ b/setup.py @@ -12,10 +12,20 @@ from nw import __version__, __url__, __docurl__, __issuesurl__, __sourceurl__ ## buildDocs = False +buildSample = False + if "qthelp" in sys.argv: buildDocs = True sys.argv.remove("qthelp") +if "sample" in sys.argv: + buildSample = True + sys.argv.remove("sample") + +## +# Qt Assistant Documentation +## + if buildDocs: buildDir = os.path.join("docs", "build", "qthelp") @@ -72,6 +82,31 @@ if buildDocs: print("Documentation build: OK") print("") +## +# Sample Project ZIP file +## + +if buildSample: + + srcSample = "sample" + dstSample = os.path.join("nw", "assets", "sample.zip") + + if os.path.isdir(srcSample): + if os.path.isfile(dstSample): + os.unlink(dstSample) + + from zipfile import ZipFile + + with ZipFile(dstSample, "w") as zipObj: + zipObj.write(os.path.join("sample", "nwProject.nwx"), "nwProject.nwx") + for docFile in os.listdir(os.path.join(srcSample, "content")): + srcDoc = os.path.join(srcSample, "content", docFile) + zipObj.write(srcDoc, "content/"+docFile) + + else: + print("Error: Could not find sample project source directory.") + sys.exit(1) + if len(sys.argv) == 1: # Nothing more to do sys.exit(0)