def register_token(session):
token = session.query(Token).first()
if token:
return token, False
url = api_method_url('project_tokens/')
headers = {'User-Agent': 'Jet Django'}
data = {'bridge_type': 'jet_bridge', 'bridge_version': VERSION}
r = requests.request('POST', url, headers=headers, data=data)
success = 200 <= r.status_code < 300
if not success:
logging.error('Register Token request error', r.status_code, r.reason)
return None, False
result = r.json()
# TODO: add serializer
token = result['token'].replace('-', '')
date_add = datetime.strptime(result['date_add'][:-6],
'%Y-%m-%dT%H:%M:%S.%f')
token = Token(token=token, date_add=date_add)
session.add(token)
session.commit()
return token, True
def set_token(session, token):
project_token = session.query(Token).first()
token_clean = str(token).replace('-', '')
if project_token:
if project_token.token == token_clean:
logging.info('This token is already set, ignoring')
return
project_token.token = token_clean
project_token.date_add = datetime.now()
session.commit()
logging.info('Token changed to {}'.format(project_token.token))
else:
project_token = Token(token=token_clean, date_add=datetime.now())
session.add(project_token)
session.commit()
logging.info('Token created {}'.format(project_token.token))