diff options
Diffstat (limited to 'src/__main__.py')
| -rw-r--r-- | src/__main__.py | 53 |
1 files changed, 50 insertions, 3 deletions
diff --git a/src/__main__.py b/src/__main__.py index 3206bc8..bd35e32 100644 --- a/src/__main__.py +++ b/src/__main__.py @@ -1,5 +1,52 @@ -# Allows for launching with python3 -m avp +from PyQt5.QtWidgets import QApplication +import sys +import logging +import re +import string -from .main import main -main() +log = logging.getLogger('AVP.Main') + + +def main() -> int: + """Returns an exit code (0 for success)""" + proj = None + mode = 'GUI' + + # Determine whether we're in GUI or commandline mode + if len(sys.argv) > 2: + mode = 'commandline' + elif len(sys.argv) == 2: + if sys.argv[1].startswith('-'): + mode = 'commandline' + else: + # remove unsafe punctuation characters such as \/?*&^%$# + sys.argv[1] = re.sub(f'[{re.escape(string.punctuation)}]', '', sys.argv[1]) + # opening a project file with gui + proj = sys.argv[1] + + # Create Qt Application + app = QApplication(sys.argv) + app.setApplicationName("audio-visualizer") + + # Launch program + if mode == 'commandline': + from .command import Command + + main = Command() + mode = main.parseArgs() + log.debug("Finished creating command object") + + # Both branches here may occur in one execution: + # Commandline parsing could change mode back to GUI + if mode == 'GUI': + from .gui.mainwindow import MainWindow + + mainWindow = MainWindow(proj) + log.debug("Finished creating MainWindow") + mainWindow.raise_() + + return app.exec_() + +if __name__ == '__main__': + sys.exit(main())
\ No newline at end of file |