def register_commandline(ip, mac):
msg = 'That Peername is already taken, please choose another. (Enter N to exit)'
if PYVER == 2:
pname = raw_input('Enter Peer Name: ') # should these be unique?
hname = raw_input('Enter Hostname: ')
elif PYVER == 3:
pname = input('Enter Peer Name: ') # should these be unique?
hname = input('Enter Hostname: ')
null, pword = utils.create_password()
added = False
while not added:
name = '%s@%s.creds' % (pname, hname)
if name not in os.listdir(os.getcwd() + '/PoolData/Creds'):
client = {
'hname': hname,
'pname': pname,
'pword': pword,
'MAC': mac,
'IP': ip,
'name': name
}
pubkey = utils.create_keys(client)
added = True
else:
if raw_input(msg).upper() == 'N':
added = True
return added, client
def put(self, *args, **kwargs):
if args or 'user' not in kwargs:
self.raise401()
username = self.get_argument('username', None)
email = self.get_argument('email', None)
password = self.get_argument('password', None)
first_name = self.get_argument('first_name', None)
last_name = self.get_argument('last_name', None)
user = kwargs['user']
update = {}
if username:
update['set__username'] = username
if email:
update['set__email'] = email
if password:
update['set__password'] = create_password(password)
if first_name:
update['set__first_name'] = first_name
if last_name:
update['set__last_name'] = last_name
try:
User.objects(username=user.username).update_one(**update)
user = User.objects(name=username or user.username).first()
self.set_status(201)
self.write(document_to_json(user, filter_set=_FILTER))
except Exception as e:
reason = e.message
self.raise400(reason=reason)
def mutate(root, info, password_form, restrictions):
print(password_form)
email = password_form['email']
password_input = password_form['password_input']
name = password_form['name']
if account_exists(email):
return CreateAccount(
ok=False, code="Error: Account with that email already exists")
if not password_valid(password_form['password_input']):
return CreateAccount(ok=False,
code="Error: Invalid password format")
password_hash = create_password(password_input)
session = uuid.uuid4().hex
params = {
'email': email,
'name': name,
'password_hash': password_hash,
'session': session,
**DietaryRestrictions.query_tag_variables(restrictions)
}
create_account_query = 'CREATE (a:Account {email: $email, name: $name, password: $password_hash, ' \
'session: $session, verified: false, completedOrientation: false, ' + DietaryRestrictions.query_tag_text() + '}), (f:Flags),' \
'(a)-[c:HasFlags]->(f) return a, c, f'
results = get_db().run(create_account_query, parameters=params)
account = None
for record in results:
account = unpack(record.get("a"))
return CreateAccount(ok=True,
code="",
session=session,
account=account)
def post(self, *args, **kwargs):
username = self.get_argument('username', None)
email = self.get_argument('email', None)
password = self.get_argument('password', None)
first_name = self.get_argument('first_name', None)
last_name = self.get_argument('last_name', None)
ip = self.get_argument('user_ip', None)
try:
password = create_password(password) if password else None
user = User(username=username, email=email,
password=password, first_name=first_name,
last_name=last_name, ip=ip).save()
self.set_status(201)
self.write(document_to_json(user, filter_set=_FILTER))
except Exception as e:
reason = e.message
self.raise400(reason=reason)
def create_user(**kwargs):
kwargs['password'] = create_password(kwargs['password'])
put_item('User', kwargs, "attribute_not_exists(email)")
def authenticate(self, user, username, password):
return user.email == username and user.password == create_password(
password)
