main.py

import argparse
import os
from getpass import getpass

import yaml
from bs4 import BeautifulSoup
from selenium.webdriver import Chrome
from selenium.webdriver.chrome.options import Options


class Manaba:

    def __init__(self, no_headless=False):
        self.attend_url = 'https://atmnb.tsukuba.ac.jp/attend/tsukuba'

        options = Options()
        if not no_headless:
            options.add_argument('--headless')
            
        try:
            self.br = Chrome(options=options)
            self.load_credentials()
        
        except:
            print('!!! Google chrome is not installed !!! ')
            exit()
            
    def load_credentials(self):
        credentials_path = os.path.expanduser('~/.manaba_attend')
        if os.environ.get('MANABA_USERNAME', '') and os.environ.get('MANABA_PASSWORD', ''):
            self.id = os.environ['MANABA_USERNAME']
            self.password = os.environ['MANABA_PASSWORD']
        elif not os.path.exists(credentials_path):
            print('Enter your credentials')
            self.id = input('id: ')
            self.password = getpass()
            with open(credentials_path, 'w') as f:
                yaml.dump({'id': self.id, 'password': self.password}, f)
            os.chmod(credentials_path, 0o600)
        else:
            with open(credentials_path) as f:
                credentials = yaml.load(f)
                self.id = credentials['id']
                self.password = credentials['password']

    def make_soup(self):
        return BeautifulSoup(self.br.page_source, 'lxml')

    def send_code(self, attend_code):
        # 1. Enter attend code
        self.br.get(self.attend_url)
        self.br.find_element_by_css_selector('[name="code"]').send_keys(attend_code + '\n')

        # attend_codeに誤りがある場合、ログイン画面には遷移しない
        s = self.make_soup()
        error = s.select('.errmsg')
        if error:
            success = False
            print(error[0].text)
            print('!!! attend failed !!!')
            return

        # 2. Login to manaba
        self.br.find_element_by_css_selector('#username').send_keys(self.id)
        self.br.find_element_by_css_selector('#password').send_keys(self.password + '\n')

        # 3. Check the state
        s = self.make_soup()
        error = s.select('.errmsg')
        if error:
            success = False
            print(error[0].text)
        else:
            body = s.select('.attend-box-body')[0].text.strip()
            description = s.select('.description')[0].text.strip()
            if '提出済' in description:
                success = True
                print(body)
            elif '提出しました' in description:
                success = True
                print(body)
            else:
                success = False
                print(body)
                print('Unknown result')

        if not success:
            print('!!! attend failed !!!')

        # 4. Close browser
        self.br.quit()


if __name__ == '__main__':
    ap = argparse.ArgumentParser()
    ap.add_argument('attend_code')
    ap.add_argument('--no-headless', action='store_true', required=False, help='specify to show the browser window.')
    args = ap.parse_args()

    manaba = Manaba(no_headless=args.no_headless)
    manaba.send_code(args.attend_code)