diff --git a/openmc_plotter/__init__.py b/openmc_plotter/__init__.py index e69de29..cd57cb1 100644 --- a/openmc_plotter/__init__.py +++ b/openmc_plotter/__init__.py @@ -0,0 +1,2 @@ + +__version__ = '0.2.0' diff --git a/openmc_plotter/__main__.py b/openmc_plotter/__main__.py index 1230416..ee8caec 100644 --- a/openmc_plotter/__main__.py +++ b/openmc_plotter/__main__.py @@ -10,10 +10,14 @@ from PySide2 import QtCore, QtGui from PySide2.QtWidgets import QApplication, QSplashScreen +from . import __version__ from .main_window import MainWindow, _openmcReload def main(): ap = ArgumentParser(description='OpenMC Plotter GUI') + version_str = f'OpenMC Plotter Version: {__version__}' + ap.add_argument('-v', '--version', action='version', version=version_str, + help='Display version info.') ap.add_argument('-d', '--model-directory', default=os.curdir, help='Location of model dir (default is current dir)') ap.add_argument('-e','--ignore-settings', action='store_false', diff --git a/setup.py b/setup.py index e1effdf..788a1b5 100644 --- a/setup.py +++ b/setup.py @@ -1,8 +1,13 @@ from setuptools import setup +# Get version information from __init__.py. This is ugly, but more reliable than +# using an import. +with open('openmc_plotter/__init__.py', 'r') as f: + version = f.readlines()[-1].split()[-1].strip("'") + kwargs = { 'name': 'openmc-plotter', - 'version': '0.2.0', + 'version': version, 'packages': ['openmc_plotter'], 'package_data': {'openmc_plotter' : ['assets/*.png']}, 'entry_points': {