Merge pull request ned14#75 from assarbad/changed-main-logic-pcmd

Rearranging how main() works (ned14#73)
willwray · Nov 18, 2022 · a75d642 · a75d642
2 parents 2ad25f1 + 62658ba
commit a75d642
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/pcpp/pcmd.py b/pcpp/pcmd.py
@@ -247,10 +247,11 @@ def on_comment(self,tok):
             return True  # Pass through
         return super(CmdPreprocessor, self).on_comment(tok)
 
-def main():
-    p = CmdPreprocessor(sys.argv)
-    sys.exit(p.return_code)
+def main(argv=None):
+    if argv is None:
+        argv = sys.argv
+    p = CmdPreprocessor(argv)
+    return p.return_code
 
 if __name__ == "__main__":
-    p = CmdPreprocessor(sys.argv)
-    sys.exit(p.return_code)
+    sys.exit(main(sys.argv))