File: custom-tfa

package info (click to toggle)
python-pyaarlo 0.8.0.15-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 556 kB
  • sloc: python: 6,064; makefile: 6; sh: 1
file content (50 lines) | stat: -rwxr-xr-x 1,401 bytes parent folder | download
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
40
41
42
43
44
45
46
47
48
49
50
#!/usr/bin/env python3
#

import logging
import os
import sys
import time

# for examples add pyaarlo install path
sys.path.append(os.path.dirname(os.path.dirname(os.path.realpath(__file__))))
import pyaarlo

# set these from the environment to log in
USERNAME = os.environ.get('ARLO_USERNAME', 'test.login@gmail.com')
PASSWORD = os.environ.get('ARLO_PASSWORD', 'test-password')

# set up logging, change INFO to DEBUG for a *lot* more information
logging.basicConfig(level=logging.DEBUG,
                    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
_LOGGER = logging.getLogger('pyaarlo')


class Arlo2FATest:
    """ 2FA authentication via console.
    Accepts input from console and returns that for 2FA.
    """

    def __init__(self):
        pass

    def start(self):
        _LOGGER.debug('2fa-cconsole: starting')
        return True

    def get(self):
        _LOGGER.debug('2fa-cconsole: checking')
        return input('Custom Enter Code: ')

    def stop(self):
        _LOGGER.debug('2fa-cconsole: stopping')


# log in
# add `verbose_debug=True` to enable even more debugging
# add `dump=True` to enable event stream packet dumps
arlo = pyaarlo.PyArlo(username=USERNAME, password=PASSWORD,
                      tfa_type='SMS', tfa_source=Arlo2FATest(),
                      save_state=False, dump=False, storage_dir='aarlo', verbose_debug=True)

time.sleep(60)