Move argparser into mains function body

pull/4/head
Darksider3 5 years ago
parent a17d0ed30f
commit 65c7bb6b3f

@ -7,12 +7,6 @@ import configparser
import lib.uis.default as default_cmd # Follows -u, -a, -f flags import lib.uis.default as default_cmd # Follows -u, -a, -f flags
default_cmd.argparser.description += " - Backups Tilde Users to stdout or a file."
args = default_cmd.argparser.parse_args()
config = configparser.ConfigParser()
config.read(args.config)
class Backup: class Backup:
filename: str filename: str
quoting: int quoting: int
@ -52,6 +46,10 @@ class Backup:
if __name__ == "__main__": if __name__ == "__main__":
default_cmd.argparser.description += " - Backups Tilde Users to stdout or a file."
args = default_cmd.argparser.parse_args()
config = configparser.ConfigParser()
config.read(args.config)
try: try:
L = ListUsers.ListUsers(config['DEFAULT']['applications_db'], L = ListUsers.ListUsers(config['DEFAULT']['applications_db'],
unapproved=args.unapproved, approved=args.approved) unapproved=args.unapproved, approved=args.approved)

@ -6,16 +6,6 @@ import configparser
import lib.UserExceptions import lib.UserExceptions
import lib.uis.config_ui # dont go to default, just following -c flag import lib.uis.config_ui # dont go to default, just following -c flag
ArgParser = lib.uis.config_ui.argparser
ArgParser.description += "- Imports a CSV file consisting of user specific details to the database"
ArgParser.add_argument('-f', '--file', default="stdout",
type=str, help='Import from CSV file', required=True)
ArgParser.add_argument('--Import', default=False, action="store_true",
help="Import Users.", required=True)
args = ArgParser.parse_args()
config = configparser.ConfigParser()
config.read(args.config)
def ImportFromFile(fname: str, db: str, userids: tuple = tuple([])): def ImportFromFile(fname: str, db: str, userids: tuple = tuple([])):
if not os.path.isfile(fname): if not os.path.isfile(fname):
@ -75,6 +65,16 @@ def ImportFromFile(fname: str, db: str, userids: tuple = tuple([])):
if __name__ == "__main__": if __name__ == "__main__":
ArgParser = lib.uis.config_ui.argparser
ArgParser.description += "- Imports a CSV file consisting of user specific details to the database"
ArgParser.add_argument('-f', '--file', default="stdout",
type=str, help='Import from CSV file', required=True)
ArgParser.add_argument('--Import', default=False, action="store_true",
help="Import Users.", required=True)
args = ArgParser.parse_args()
config = configparser.ConfigParser()
config.read(args.config)
try: try:
if not args.Import: if not args.Import:
print("Error, need the import flag") print("Error, need the import flag")

@ -8,35 +8,34 @@ import lib.System
import lib.UserExceptions import lib.UserExceptions
import sqlite3 import sqlite3
lib.uis.config_ui.argparser.description += " - Edit Tilde Users"
ArgParser = lib.uis.config_ui.argparser if __name__ == "__main__":
ArgParser.add_argument('--user', type=str, lib.uis.config_ui.argparser.description += " - Edit Tilde Users"
ArgParser = lib.uis.config_ui.argparser
ArgParser.add_argument('--user', type=str,
help='Tilde users name to edit', required=True) help='Tilde users name to edit', required=True)
Mutually = ArgParser.add_mutually_exclusive_group() Mutually = ArgParser.add_mutually_exclusive_group()
Mutually.add_argument('-r', '--remove', default=False, action="store_true", Mutually.add_argument('-r', '--remove', default=False, action="store_true",
help='Remove an approved/unapproved User from the system(and DB). Effectively purges him.', help='Remove an approved/unapproved User from the system(and DB). Effectively purges him.',
required=False) required=False)
Mutually.add_argument("--verify", default=True, action="store_false", Mutually.add_argument("--verify", default=True, action="store_false",
help="Turns off value checks", help="Turns off value checks",
required=False) required=False)
ArgParser.add_argument('--sshpubkey', type=str, default=None, ArgParser.add_argument('--sshpubkey', type=str, default=None,
help="Stores the new given SSH-Key in given user", required=False) help="Stores the new given SSH-Key in given user", required=False)
ArgParser.add_argument('--name', type=str, default=None, ArgParser.add_argument('--name', type=str, default=None,
help="Sets the stored name of the given user") help="Sets the stored name of the given user")
ArgParser.add_argument('--username', type=str, default=None, ArgParser.add_argument('--username', type=str, default=None,
help="Rename given User") help="Rename given User")
ArgParser.add_argument('--email', type=str, default=None, ArgParser.add_argument('--email', type=str, default=None,
help="Set new email address for given user") help="Set new email address for given user")
ArgParser.add_argument('--status', type=int, default=None, ArgParser.add_argument('--status', type=int, default=None,
help="Set status of given user") help="Set status of given user")
args = ArgParser.parse_args() args = ArgParser.parse_args()
config = configparser.ConfigParser() config = configparser.ConfigParser()
config.read(args.config) config.read(args.config)
if __name__ == "__main__":
try: try:
db = config['DEFAULT']['applications_db'] db = config['DEFAULT']['applications_db']
if not args.sshpubkey and not args.name and not args.username and not args.email and args.status is None \ if not args.sshpubkey and not args.name and not args.username and not args.email and args.status is None \

Loading…
Cancel
Save