42 lines
1.0 KiB
Python
42 lines
1.0 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import argparse
|
||
|
|
import sys
|
||
|
|
import os
|
||
|
|
# This is ugly as fuck. TODO: can we do this more cleanly?
|
||
|
|
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), os.path.pardir)))
|
||
|
|
import config
|
||
|
|
|
||
|
|
class DBmgr(object):
|
||
|
|
def __init__(self, args = None):
|
||
|
|
self.DB = config.DB
|
||
|
|
self.args = args
|
||
|
|
|
||
|
|
def keyChk(self):
|
||
|
|
# Is it a pubkey file?
|
||
|
|
if os.path.isfile(os.path.abspath(os.path.expanduser(self.args['key']))):
|
||
|
|
with open(os.path.abspath(os.path.expanduser(self.args['key'])), 'r') as f:
|
||
|
|
self.args['key'] = f.read()
|
||
|
|
self.args['key'] = self.args['key'].strip()
|
||
|
|
|
||
|
|
|
||
|
|
def add(self, key, host, role):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def argParse():
|
||
|
|
args = argparse.ArgumentParser()
|
||
|
|
args.add_argument('-k',
|
||
|
|
'--key',
|
||
|
|
dest = 'key',
|
||
|
|
default = None,
|
||
|
|
type = 'str',
|
||
|
|
|
||
|
|
return(args)
|
||
|
|
|
||
|
|
def main():
|
||
|
|
args -
|
||
|
|
d = DBmgr(args)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|