time_table.py

#!/usr/local/bin/python3
# -*- coding: utf-8 -*-
import csv
import datetime
import re

import setting as fs
import sqlite3
import logging
from data_objects import *

# 与えられた日付の時間割を返す
# 日付が不正ならNoneを返す
# 休みなどで時間割がなければ空リストを返す
def get_time_table(date):
    date = get_date(date)
    try:
        input_date = datetime.datetime.strptime(date, '%Y/%m/%d')
    except ValueError:
        print("誤った日付です")
        return None

    # デフォルトの時間割を読み込む
    with open(fs.TIME_TABLE, 'r', encoding = 'utf-8') as f:
        table = []
        reader = csv.reader(f)
        for row in reader:
            table.append(row)

    # 時間割変更を読み込む
    change_list = get_time_table_change(date)

    # イベントを読み込む
    event_list = get_event(date)

    week = {'月曜': 0, '火曜': 1, '水曜': 2, '木曜': 3, '金曜': 4}
    table_event = ''

    for e in event_list:
        if e.event == '休み' or e.event == '時間割なし':
            table_event = '時間割なし'
            break
        for d in week.keys():
            if e.event == d:
                table_event = e.event
                break
    week_point = input_date.weekday()

    if table_event == '時間割なし' or week_point >= 5:
        return []
    elif table_event != '':
        week_point = week[table_event]

    for e in change_list:
        times = re.split('[.,]', e.time)
        for t in times:
            table[week_point][int(t)-1] = e.subject

    return table[week_point]


# 要求された日付にある課題を返す(list<Task>)
# なければ空リスト
def get_task(date):
    tasks = access("SELECT * FROM task WHERE date = '{}'".format(get_date(date)))
    out = list()
    for task in tasks:
        out.append(Task(task))
    return out


# 要求された日付以降にある課題を返す(list<list<Task>>)
# なければ空リスト
def get_task_list(date):
    tasks = get_task_dict(date)
    out = list()
    for d in sorted(tasks.keys()):
        out.append(tasks[d])
    return out


# 要求された日付以降にある課題を返す(list<list<Task>>)
# なければ空リスト
def get_task_dict(date):
    date_list = access("SELECT DISTINCT date FROM task WHERE date >= '{}'".format(get_date(date)))
    task_dict = dict()
    for d in date_list:
        task_dict[d[0]] = list()
    tasks = access("SELECT * FROM task WHERE date >= '{}' ORDER BY date ASC ".format(get_date(date)))
    for task in tasks:
        task_dict[task[0]].append(Task(task))
    return task_dict


# 要求された日付以降にある課題を返す(list<Task>)
# なければ空リスト
def get_task_list_one(date):
    tasks = access("SELECT * FROM task WHERE date >= '{}' ORDER BY date ASC ".format(get_date(date)))
    out = list()
    for task in tasks:
        out.append(Task(task))
    return out


# 要求された日付にあるイベントを返す(list<Event>)
# なければ空リスト
def get_event(date):
    events = access("SELECT * FROM event WHERE date = '{}'".format(get_date(date)))
    out = list()
    for event in events:
        out.append(Event(event))
    return out


# 要求された日付以降にある課題を返す(list<list<Task>>)
# なければ空リスト
def get_event_dict(date):
    event_list = access("SELECT DISTINCT date FROM event WHERE date >= '{}'".format(get_date(date)))
    event_dict = dict()
    for d in event_list:
        event_dict[d[0]] = list()
    events = access("SELECT * FROM event WHERE date >= '{}' ORDER BY date ASC ".format(get_date(date)))
    for event in events:
        event_dict[event[0]].append(Event(event))
    return event_dict


# 要求された日付以降にあるイベントを返す(list<list<Event>>)
# なければ空リスト
def get_event_list(date):
    events = get_event_dict(date)
    out = list()
    for d in sorted(events.keys()):
        out.append(events[d])
    return out


# 要求された日付以降にあるイベントを返す(list<Event>)
# なければ空リスト
def get_event_list_one(date):
    events = access("SELECT * FROM event WHERE date >= '{}' ORDER BY date ASC ".format(get_date(date)))
    out = list()
    for event in events:
        out.append(Event(event))
    return out


# time_table_change内にある変更を返す(list<Change>)
# なければ空リスト
def get_time_table_change(date):
    changes = access("SELECT * FROM time_table_change WHERE date = '{}'".format(get_date(date)))
    out = list()
    for change in changes:
        out.append(Change(change))
    return out


# 要求された日付以降にある時間割変更を返す(list<list<Change>>)
# なければ空リスト
def get_time_table_change_dict(date):
    event_list = access("SELECT DISTINCT date FROM time_table_change WHERE date >= '{}'".format(get_date(date)))
    event_dict = dict()
    for d in event_list:
        event_dict[d[0]] = list()
    changes = access("SELECT * FROM time_table_change WHERE date >= '{}' ORDER BY date ASC ".format(get_date(date)))
    for change in changes:
        event_dict[change[0]].append(Change(change))
    return event_dict


# 要求された日付以降にあるイベントを返す(list<list<Event>>)
# なければ空リスト
def get_time_table_change_list(date):
    changes = get_time_table_change_dict(date)
    out = list()
    for d in sorted(changes.keys()):
        out.append(changes[d])
    return out


# 要求された日付以降にあるイベントを返す(list<Event>)
# なければ空リスト
def get_time_table_change_list_one(date):
    chages = access("SELECT * FROM time_table_change WHERE date >= '{}' ORDER BY date ASC ".format(get_date(date)))
    out = list()
    for event in chages:
        out.append(Change(event))
    return out


# 時間割変更をdbに追記する
# 成功ならTrueを返す
def add_time_table_change(date, time, subject):
    try:
        access('INSERT INTO time_table_change(date, time, subject) VALUES {}'.format((get_date(date), time, subject)))
        return True
    except Exception:
        return False


# イベントをdbに追記する
# 成功ならTrueを返す
def add_event(date, event):
    try:
        access("INSERT INTO event(date, event) VALUES {}".format((get_date(date), event)))
        return True
    except Exception:
        return False


# 課題をdbに追記する
# 成功ならTrueを返す
def add_task(date, subject, value):
    try:
        access('INSERT INTO task(date, subject, value) VALUES {}'.format((get_date(date), subject, value)))
        return True
    except Exception:
        return False


# 時間割変更をdbから削除する
# 成功ならTrueを返す
def delete_time_table_change(date, time, subject):
    try:
        access("DELETE FROM time_table_change WHERE date == '{}' AND time == '{}' AND subject == '{}'".format(get_date(date), time, subject))
        return True
    except Exception:
        return False


# イベントをdbから削除する
# 成功ならTrueを返す
def delete_event(date, event):
    try:
        access("DELETE FROM event WHERE date == '{}' AND event == '{}'".format(get_date(date), event))
        return True
    except Exception:
        return False


# 課題をdbから削除する
# 成功ならTrueを返す
def delete_task(date, subject, value):
    try:
        access("DELETE FROM task WHERE date == '{}' AND subject == '{}' AND value == '{}'".format(get_date(date), subject, value))
        return True
    except Exception:
        return False


# 指定IDの時間割変更をdbから削除する
# 成功ならTrueを返す
def delete_time_table_change_id(id):
    try:
        access("DELETE FROM time_table_change WHERE id == {}".format(id))
        return True
    except Exception:
        return False


# 指定IDのイベントをdbから削除する
# 成功ならTrueを返す
def delete_event_id(id):
    try:
        access("DELETE FROM event WHERE id == {}".format(id))
        return True
    except Exception:
        return False


# 指定IDの課題をdbから削除する
# 成功ならTrueを返す
def delete_task_id(id):
    try:
        access("DELETE FROM task WHERE id == {}".format(id))
        return True
    except Exception:
        return False


# 指定IDの時間割変更を更新する
# 成功ならTrueを返す
def update_time_table_change(id, date, time, subject):
    try:
        access("UPDATE time_table_change SET date='{}', time='{}', subject='{}' WHERE id = {}".format(get_date(date), time, subject, id))
        return True
    except Exception:
        return False


# 指定IDのイベントを更新する
# 成功ならTrueを返す
def update_event(id, date, event):
    try:
        access("UPDATE event SET date='{}', event='{}' WHERE id = {}".format(get_date(date), event, id))
        return True
    except Exception:
        return False


# 指定IDの課題を更新する
# 成功ならTrueを返す
def update_task(id, date, subject, value):
    try:
        access("UPDATE task SET date='{}', subject='{}', value='{}' WHERE id = {}".format(get_date(date), subject, value, id))
        return True
    except Exception:
        return False


# 引数の日付の正当性を確かめる
def check_date(date):
    try:
        datetime.datetime.strptime(date, '%Y/%m/%d')
        return True
    except ValueError:
        print("誤った日付です")
        return False


# 今日や明日などの文字列を日付に変換する
# すでに日付ならばそのまま返す
def get_date(date_name):
    if date_name == '今日':
        return datetime.datetime.today().strftime('%Y/%m/%d')
    if date_name == '明日':
        return (datetime.datetime.today() + datetime.timedelta(days=1)).strftime('%Y/%m/%d')
    if date_name == '明後日':
        return (datetime.datetime.today() + datetime.timedelta(days=2)).strftime('%Y/%m/%d')
    if date_name == '明々後日':
        return (datetime.datetime.today() + datetime.timedelta(days=3)).strftime('%Y/%m/%d')
    if date_name == '昨日':
        return (datetime.datetime.today() + datetime.timedelta(days=-1)).strftime('%Y/%m/%d')
    if date_name == '来週':
        return (datetime.datetime.today() + datetime.timedelta(days=7)).strftime('%Y/%m/%d')
    return format_date(date_name)


# 与えられた日付を 年4桁/月2桁/日2桁(str) の形式に変換する
def format_date(date):
    today = datetime.date.today()
    d = date.split('/')
    if len(d) == 1:
        if int(date) < int(today.strftime('%d')):
            d.insert(0, str(int(today.strftime('%m'))+1))
        else:
            d.insert(0, today.strftime('%m'))
    if len(d) == 2:
        if int(d[0]) < int(today.strftime('%m')) or int(d[0]) == int(today.strftime('%m')) and int(d[1]) < int(today.strftime('%d')):
            d.insert(0, str(int(today.strftime('%Y'))+1))
        else:
            d.insert(0, today.strftime('%Y'))
    if len(d) == 3:
        month = d[1] if len(d[1]) == 2 else '0' + d[1]
        day = d[2] if len(d[2]) == 2 else '0' + d[2]
        date_s = f'{d[0]}/{month}/{day}'
        if check_date(date_s):
            return date_s
        raise ValueError


def access(query):
    logger = logging.getLogger('sql')
    f = logging.FileHandler('./bot_log/sql-access-'+datetime.datetime.today().strftime('%Y-%m')+'.log')
    f.setFormatter(logging.Formatter('%(asctime)s: %(message)s'))
    logger.addHandler(f)
    logger.log(30, query)
    connection = sqlite3.connect(fs.SQL)
    cursor = connection.cursor()
    result = cursor.execute(query).fetchall()
    connection.commit()
    cursor.close()
    connection.close()
    return result


def data_base_initialize():
    access('''CREATE TABLE task(date text NOT NULL, subject text NOT NULL, value text NOT NULL, id integer PRIMARY KEY NOT NULL, UNIQUE(date, subject, value))''')
    access('''CREATE TABLE event(date text NOT NULL, event text NOT NULL, id integer PRIMARY KEY NOT NULL, UNIQUE(date, event))''')
    access('''CREATE TABLE time_table_change(date text NOT NULL, time text NOT NULL, subject text NOT NULL, id integer PRIMARY KEY NOT NULL, UNIQUE(date, time, subject))''')

if __name__ == '__main__':
    data_base_initialize()