102 lines
3.2 KiB
Plaintext
102 lines
3.2 KiB
Plaintext
|
#!/bin/python3
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
import logging
|
||
|
|
import requests
|
||
|
|
import json
|
||
|
|
import argparse
|
||
|
|
import datetime
|
||
|
|
|
||
|
|
logging.basicConfig(format='%(levelname)s\t- %(message)s')
|
||
|
|
logger = logging.getLogger()
|
||
|
|
logger.setLevel(logging.INFO)
|
||
|
|
|
||
|
|
state = {}
|
||
|
|
|
||
|
|
def validDate(date):
|
||
|
|
try:
|
||
|
|
datetime.datetime.strptime(input,"%d/%m/%Y")
|
||
|
|
except ValueError as err:
|
||
|
|
logger.error("Date is invalid. Input format: DD/MM/YYYY")
|
||
|
|
sys.exit(-1)
|
||
|
|
|
||
|
|
#session file - [-s username] [-d nt/ot/et date]
|
||
|
|
def list_docs(args):
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
|
||
|
|
parser.add_argument('session', nargs='?', default=None)
|
||
|
|
parser.add_argument("-s", '--username', nargs=1, help="Username")
|
||
|
|
parser.add_argument("-dnt", '--newerThan', help="Date new than")
|
||
|
|
parser.add_argument("-dot", '--olderThan', help="Date older than")
|
||
|
|
parser.add_argument("-deq", '--equalTo', help="Date equal to")
|
||
|
|
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
# Check if session file is valid
|
||
|
|
if args.session:
|
||
|
|
if (not os.path.isfile(args.session)):
|
||
|
|
logger.error("File '" + args.session + "' not found.")
|
||
|
|
sys.exit(-1)
|
||
|
|
else:
|
||
|
|
logger.error("Need session file.")
|
||
|
|
sys.exit(-1)
|
||
|
|
|
||
|
|
if args.newerThan:
|
||
|
|
validDate(args.newerThan)
|
||
|
|
|
||
|
|
try:
|
||
|
|
subjects = requests.get(f'http://{state['REP_ADDRESS']}/user/list',
|
||
|
|
json=json.dumps({'username' : args.username, 'newerThan' : args.newerThan}))
|
||
|
|
subjects.raise_for_status()
|
||
|
|
|
||
|
|
except requests.exceptions.RequestException as errex:
|
||
|
|
logger.error("Failed to obtain response from server.")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
subjects = json.loads(subjects)
|
||
|
|
|
||
|
|
elif args.equalTo:
|
||
|
|
validDate(args.equalTo)
|
||
|
|
|
||
|
|
try:
|
||
|
|
subjects = requests.get(f'http://{state['REP_ADDRESS']}/user/list',
|
||
|
|
json=json.dumps({'username' : args.username, 'equalTo' : args.equalTo}))
|
||
|
|
subjects.raise_for_status()
|
||
|
|
|
||
|
|
except requests.exceptions.RequestException as errex:
|
||
|
|
logger.error("Failed to obtain response from server.")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
subjects = json.loads(subjects)
|
||
|
|
|
||
|
|
elif args.olderThan:
|
||
|
|
validDate(args.olderThan)
|
||
|
|
|
||
|
|
try:
|
||
|
|
subjects = requests.get(f'http://{state['REP_ADDRESS']}/user/list',
|
||
|
|
json=json.dumps({'username' : args.username, 'olderThan' : args.olderThan}))
|
||
|
|
subjects.raise_for_status()
|
||
|
|
|
||
|
|
except requests.exceptions.RequestException as errex:
|
||
|
|
logger.error("Failed to obtain response from server.")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
subjects = json.loads(subjects)
|
||
|
|
|
||
|
|
else:
|
||
|
|
try:
|
||
|
|
subjects = requests.get(f'http://{state['REP_ADDRESS']}/user/list')
|
||
|
|
subjects.raise_for_status()
|
||
|
|
|
||
|
|
except requests.exceptions.RequestException as errex:
|
||
|
|
logger.error("Failed to obtain response from server.")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
subjects = json.loads(subjects)
|
||
|
|
|
||
|
|
for s in subjects:
|
||
|
|
sys.stdout.write(s['id'] + " - " + s['username'])
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
list_docs(sys.argv[1:])
|