qwer.py

#!/usr/bin/env python3

import os
import sys
import csv
import time
import random
import string
import getopt
import signal
import pathlib
import datetime
import urllib.parse
from typing import List
from random import randrange


# date:   2022-07-08
# author: duruyao@gmail.com
# desc:   command line game for typing practice


def debug_ln(values, end='\n'):
    print('\033[1;32;35m', values, '\033[m', sep='', end=end)


def info_ln(values, end='\n'):
    print('\033[1;32;32m', values, '\033[m', sep='', end=end)


def trace_ln(values, end='\n'):
    print('\033[1;32;34m', values, '\033[m', sep='', end=end)


def warning_ln(values, end='\n'):
    print('\033[1;32;33m', values, '\033[m', sep='', end=end)


def error_ln(values, end='\n'):
    print('\033[1;32;31m', values, '\033[m', sep='', end=end)


def show_usage():
    """Usage: {execute} [OPTIONS]

Command line game for typing practice

Options:
  -b, --batch=<N>                       Set batch size (default: 20)
  -d, --diff=<easy|normal|hard|ielts>   Select game difficulty (default: ielts)
  -l, --length=<N>                      Set the length of random string while diff != ielts (default: 1)
  -h, --help                            Display this help message

Examples:
  {execute}
  {execute} -b 20 -d ielts
  {execute} --batch=20 --diff=ielts
  {execute} -l 1 -b 20 -d easy
  {execute} --length=1 --batch=20 --diff=easy
    """
    execute = 'qwer' if '.py' not in sys.argv[0] else f'python3 {os.path.realpath(sys.argv[0])}'
    print(show_usage.__doc__.format(execute=execute))


def get_ielts_words() -> List[List[str]]:
    result = []
    new_url_fmt = 'https://dict.youdao.com/result?word={}&lang=en'
    with open(f'{os.path.dirname(os.path.realpath(sys.argv[0]))}/doc/ielts_words.csv', 'r') as file:
        reader = csv.DictReader(file, delimiter=',')
        for row in reader:
            result.append([row['WORD'], row['US PRONUNCIATION'],
                           row['TRANSLATE'], new_url_fmt.format(urllib.parse.quote(row['WORD']))])
    return result


def rand_str(chars=string.ascii_lowercase + ',./;', length=10, sep=''):
    return sep.join(random.choice(chars) for _ in range(length))


def signal_handler(sig, frame):
    print('\n\nExit The Game ...')
    sys.exit(0)


def main():
    # handle CTRL_C
    signal.signal(signal.SIGINT, signal_handler)

    # set default values
    diff = 'ielts'
    length = 1
    batch = 20
    thesaurus = {'easy': string.ascii_lowercase + string.ascii_lowercase + ',.?;:"',
                 'normal': string.ascii_lowercase + string.ascii_letters + string.digits + ',.?;:"',
                 'hard': (string.ascii_lowercase + string.ascii_lowercase + string.ascii_letters +
                          string.digits + string.punctuation)}
    ielts_words = get_ielts_words()

    # parse arguments
    try:
        opts, args = getopt.getopt(sys.argv[1:], "b:d:l:h", ["batch=", "diff=", "length=", "help"])
    except getopt.GetoptError as e:
        error_ln(e)
        show_usage()
        sys.exit(1)
    for opt_key, opt_value in opts:
        if opt_key in ('-b', '--batch'):
            batch = int(opt_value)
        elif opt_key in ('-d', '--diff'):
            diff = str(opt_value)
        elif opt_key in ('-l', '--length'):
            length = int(opt_value)
        elif opt_key in ('-h', '--help'):
            show_usage()
            sys.exit(0)

    length = 0 if diff == 'ielts' else length
    level = '{}-{}-{}'.format(diff, length, batch)
    history_dir = '{}/.qwer'.format(os.path.expanduser('~'))
    history_file = '{}/{}.md'.format(history_dir, level)

    pathlib.Path(history_dir).mkdir(parents=True, exist_ok=True)
    if not os.path.exists(history_file):
        with open(history_file, 'w') as file:
            file.write('| SCORE  | LEVEL          | DATE                | GAMER            |\n')
            file.write('|--------|----------------|---------------------|------------------|\n')

    # set default values
    gain = 0
    loss = 0
    gamer = 'someone'

    for i in 3, 2, 1, 'Go':
        print('{:<2}!'.format(i), sep='')
        time.sleep(1)
    print()

    for i in range(batch):
        print('[{:>3}]'.format(i + 1))
        word, us, trans, url = ielts_words[randrange(len(ielts_words))] if diff == 'ielts' \
            else [rand_str(chars=thesaurus[diff], length=length), '', '', '']
        answer = ''
        begin = datetime.datetime.now()
        while answer != word:
            print('      Q: ', end='')
            trace_ln('{:<20}'.format(word), end=' ')
            debug_ln(f'{us} {trans} {url}')
            print('      A: ', end='')
            answer = input()
            us, trans, url = ['', '', '']
        end = datetime.datetime.now()
        duration = end - begin
        print('      T: ', end='')
        if duration.total_seconds() < len(word) + 0.5:
            info_ln(duration)
            gain += 1
        else:
            warning_ln(duration)
            loss += 1
    score = '%.2f' % ((gain * 100) / (gain + loss))

    print()
    now = datetime.datetime.now()
    date = '{:>4}-{:0>2}-{:0>2} {:0>2}:{:0>2}:{:0>2}'. \
        format(now.year, now.month, now.day, now.hour, now.minute, now.second)
    print('SCORE: %s' % score)
    print('YOUR NAME: ', end='')
    name = input()
    if len(name):
        gamer = name[:16]
    print('HISTORY: ')

    with open(history_file, 'r') as file:
        lines = file.readlines()
    lines.append('| {:0>6} | {:<14} | {:<19} | {:<16} | <-\n'.format(score, level, date, gamer))

    for line in lines[:2]:
        print('        ' + line, end='')
    lines[:] = lines[2:]
    lines.sort(reverse=True)
    for line in lines:
        print('        ' + line, end='')

    with open(history_file, 'a') as file:
        file.write('| {:0>6} | {:<14} | {:<19} | {:<16} |\n'.format(score, level, date, gamer))


if __name__ == '__main__':
    main()