FilesExpand file tree

 master

/

auth.py

Copy path

BlameMore file actions

BlameMore file actions

Latest commit

 

History

History

History

39 lines (33 loc) · 1.33 KB

 master

/

auth.py

Top

File metadata and controls

• Code
• Blame

39 lines (33 loc) · 1.33 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

"""

A module dedicated to authenticating users via Deepcode website

"""

import asyncio

import webbrowser

import aiohttp

from .utils import logger

from .connection import api_call

from .constants import AUTH_POLLING_INTERVAL, DEFAULT_SOURCE

async def login(service_url, source=DEFAULT_SOURCE):

"""

Initiate a new login protocol.

User will be forwarded to Deepcode website to complete the process.

"""

res = await api_call('login', method="POST", data={'source': source})

api_key, login_url = res['sessionToken'], res['loginURL']

# Open browser to complete authentication and assign api_key and user

webbrowser.open_new_tab(login_url)

while(True):

try:

await asyncio.sleep(AUTH_POLLING_INTERVAL)

status = await api_call('session', method="GET",

callback=asyncio.coroutine(lambda r: r.status),

api_key=api_key)

if status == 304:

print('Please, complete login process in opened browser. Re-checking session in {} sec'.format(AUTH_POLLING_INTERVAL))

continue

else:

# it means success, when we got 200 status code

return api_key

except aiohttp.client_exceptions.ClientResponseError:

logger.error('Missing or invalid sessionToken')

raise