diff --git a/epilogue/__init__.py b/epilogue/__init__.py index f900313..124c534 100644 --- a/epilogue/__init__.py +++ b/epilogue/__init__.py @@ -15,8 +15,18 @@ ARCHIVES_BASEDIR = "archives" FORUM_ARCHIVES = os.path.join(ARCHIVES_BASEDIR, "forums") WIKI_ARCHIVES = os.path.join(ARCHIVES_BASEDIR, "wiki") +DEFAULT_ARGUMENTS = ["forum"] # ["wiki", "forum"] + def main(): - archive_forum() + args = sys.argv[1:] + if not args: + args = DEFAULT_ARGUMENTS + + if "forum" in args: + archive_forum() + + if "wiki" in args: + archive_wiki() def archive_forum(): forum = Forum(FORUM_DATABASE)