def loadOrgs(add, file):
if not os.path.isfile(file):
print 'file not found ', file
sys.exit(1)
fobj = open(file, 'r')
reader = csv.reader(fobj)
for row in reader:
if add:
print 'add ', row
datawake_mysql.addOrgLink(row[0], row[1])
else:
print 'delete ', row
datawake_mysql.deleteOrgLink(row[0], row[1])
fobj.close()
def loadOrgs(add, file):
if not os.path.isfile(file):
print 'file not found ', file
sys.exit(1)
fobj = open(file, 'r')
reader = csv.reader(fobj)
for row in reader:
if add:
print 'add ', row
datawake_mysql.addOrgLink(row[0], row[1])
else:
print 'delete ', row
datawake_mysql.deleteOrgLink(row[0], row[1])
fobj.close()
"""
def loadOrgs(add, file):
if not os.path.isfile(file):
print 'file not found ', file
sys.exit(1)
fobj = open(file, 'r')
reader = csv.reader(fobj)
for row in reader:
if add:
print 'add ', row
datawake_mysql.addOrgLink(row[0], row[1])
else:
print 'delete ', row
datawake_mysql.deleteOrgLink(row[0], row[1])
fobj.close()
if __name__ == '__main__':
parser = argparse.ArgumentParser(description='Add or remove email org links')
parser.add_argument('orgFile', help='csv file of email address, org name')
parser.add_argument('--delete', action='store_false', help='delete relationships. default=add')
parser.add_argument('--demo', action='store_true', help="Set up demo org")
args = parser.parse_args()
if args.demo:
datawake_mysql.addOrgLink("*****@*****.**", "MEMEXDEMO")
else:
loadOrgs(args.delete, args.orgFile)
def loadUser(user,org,add):
if add:
datawake_mysql.addOrgLink(user, org)
else:
datawake_mysql.deleteOrgLink(user, org)
def loadUser(user, org, add):
if add:
datawake_mysql.addOrgLink(user, org)
else:
datawake_mysql.deleteOrgLink(user, org)
fobj.close()
def loadUser(user, org, add):
if add:
datawake_mysql.addOrgLink(user, org)
else:
datawake_mysql.deleteOrgLink(user, org)
if __name__ == '__main__':
parser = argparse.ArgumentParser(
description='Add or remove email org links')
parser.add_argument('--file', help='csv file of email address, org name')
parser.add_argument('--delete',
action='store_false',
help='delete relationships. default=add')
parser.add_argument('--demo', action='store_true', help="Set up demo org")
parser.add_argument('--user', help='user email address to add')
parser.add_argument('--org', help='users org')
args = parser.parse_args()
if args.demo:
datawake_mysql.addOrgLink("*****@*****.**", "MEMEXDEMO")
elif args.file:
loadOrgs(args.delete, args.orgFile)
elif args.user and args.org:
loadUser(args.user, args.org, args.delete)
else:
print 'invalid args: try --help'
sys.exit(1)
