from selenium import webdriver
from selenium.webdriver import ChromeOptions 
from selenium.webdriver.chrome.webdriver import WebDriver
import csv
import io
import os
import sys
import time

WAIT_TIME = 5

sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8')
item_code = None


def login(driver):
    driver.get('https://www.sbisec.co.jp/ETGate')

    loginid = driver.find_element_by_name('user_id').send_keys('userid')
    passwd = driver.find_element_by_name('user_password').send_keys('password')

    Log_in = driver.find_element_by_name('ACT_login').click()


def download_csvfile(driver, year, month, day):
    driver.get('https://site2.sbisec.co.jp/ETGate/?_ControlID=WPLETacR007Control')
    From_year = driver.find_element_by_name('ref_from_yyyy').send_keys(year)
    From_month = driver.find_element_by_name('ref_from_mm').send_keys(month)
    From_day = driver.find_element_by_name('ref_from_dd').send_keys(day)

    Search = driver.find_element_by_name('ACT_search').click()
    Download_csv = driver.find_element_by_id('csvlink').click()


def read_data(input_csv):
    csv_data = []
    with open(input_csv, newline='', encoding='shift-jis') as f:

        # pass head rows
        for (n, h) in enumerate(csv.DictReader(f), start=1):
            if n >= 5:
                break

        input_csv_header = ['約定日', '銘柄', '銘柄コード', '市場', '取引', '期限', '預り', '課税',
                            '約定数量', '約定単価', '手数料/諸経費等', '税額', '受渡日',
                            '受渡金額/決済損益']

        reader = csv.DictReader(f, input_csv_header)
        for row in reader:
            if item_code is None:
                # pass the exception rows of stock trade
                if row['取引'] != '株式現物買' and row['取引'] != '株式現物売':
                    continue
            else:
                if row['銘柄コード'] != item_code or (row['取引'] != '株式現物買' and row['取引'] != '株式現物売'):
                    continue

            csv_data.append(row)

    return csv_data


def calculate_data(csv_data):
    output_data = []
    for i, row in enumerate(csv_data):
        amount = 0
        price = 0
        for j in range(i + 1, len(csv_data)):
            if row['銘柄コード'] == csv_data[j]['銘柄コード']:
                brand = csv_data[j]['銘柄']
                brand_code = csv_data[j]['銘柄コード']

                if row['取引'] == '株式現物買':
                    sell_day = csv_data[j]['約定日']
                    buy_day = row['約定日']
                    sell_price = csv_data[j]['約定単価']
                    buy_price = row['約定単価']
                    sell_amount = csv_data[j]['約定数量']
                    buy_amount = row['約定数量']

                    if amount == 0:
                        amount = int(buy_amount) - int(sell_amount)
                        price = float(csv_data[j]['受渡金額/決済損益']) - float(row['受渡金額/決済損益'])
                    else:
                        amount -= int(sell_amount)
                        price += float(csv_data[j]['受渡金額/決済損益'])

                    if amount == 0:
                        item = {'現買日': buy_day,
                                '現売日': sell_day,
                                '銘柄': brand,
                                '銘柄コード': brand_code,
                                '購入数量': buy_amount,
                                '現買単価': buy_price,
                                '現売単価': sell_price,
                                '決済損益': price}

                else:
                    buy_day = csv_data[j]['約定日']
                    sell_day = row['約定日']
                    buy_price = csv_data[j]['約定単価']
                    sell_price = row['約定単価']
                    buy_amount = csv_data[j]['約定数量']
                    sell_amount = row['約定数量']

                    if amount == 0:
                        amount = int(buy_amount) - int(sell_amount)
                        price = float(row['受渡金額/決済損益']) - float(csv_data[j]['受渡金額/決済損益'])
                    else:
                        amount += int(buy_amount)
                        price -= float(csv_data[j]['受渡金額/決済損益'])

                    if amount == 0:
                        item = {'現買日': buy_day,
                                '現売日': sell_day,
                                '銘柄': brand,
                                '銘柄コード': brand_code,
                                '購入数量': buy_amount,
                                '現買単価': buy_price,
                                '現売単価': sell_price,
                                '決済損益': price}

                if amount == 0:
                    csv_data.pop(j)
                    output_data.append(item)
                    break

    return output_data


def write_data(output_data):
    output_csv = 'output.csv'
    output_csv_header = ['現買日', '現売日', '銘柄', '銘柄コード', '購入数量', '現買単価',
                        '現売単価', '決済損益']

    with open(output_csv, 'w', newline='', encoding='utf-8') as f:
        writer = csv.DictWriter(f, fieldnames=output_csv_header)
        writer.writeheader()
        total = 0
        for row in output_data:
            writer.writerow(row)
            total += float(row['決済損益'])
        print('\"total record: {}'.format(len(output_data)))
        print('total gain/loss: {}'.format(total))

        top_list = {}
        top_list = sorted(output_data, key=lambda x: x['決済損益'], reverse=True)
        bottom_list = {}
        bottom_list = sorted(output_data, key=lambda x: x['決済損益'])

        print('\n=TOP5=')
        for i in range(5 if len(top_list) >= 5 else len(top_list)):
            print(top_list[i])
        print('\n=BOTTOM5=')
        for i in range(5 if len(bottom_list) >= 5 else len(bottom_list)):
            print(bottom_list[i])
        # quotation
        print('\"')


def main():
    # setting folder for downloading
    downloadsFilePath = '.\\'
    options = ChromeOptions()
    prefs = {
                "profile.default_content_settings.popups": 1,
                "download.default_directory": 
                        os.path.abspath(downloadsFilePath) + r"\\",
                "directory_upgrade": True
            }
    options.add_experimental_option("prefs", prefs)

    driver = webdriver.Chrome(executable_path=r'./chromedriver.exe', options=options)
    driver.implicitly_wait(WAIT_TIME)

    login(driver)

    Account = driver.find_element_by_xpath('//a[img/@title="口座管理"]').click()

    # download csv file from 2021/01/01
    download_csvfile(driver, '2021', '01', '01')
    time.sleep(WAIT_TIME)

    # get the latest csv file name
    filename = None
    if len(os.listdir(downloadsFilePath)) != 0:
        filename = max([downloadsFilePath + '\\'+ f for f in os.listdir(downloadsFilePath) if f.endswith('.csv')], key=os.path.getctime)

    driver.close()

    csv_data = read_data(filename)

    output_data = calculate_data(csv_data)

    write_data(output_data)


if __name__== '__main__':
    main()