from .tool.func import *

async def vote_end(num = 1):
    num = str(num)
    
    with get_db_connect() as conn:
        curs = conn.cursor()

        curs.execute(db_change('select name, subject, data, type from vote where id = ? and user = ""'), [num])
        data_list = curs.fetchall()
        if not data_list:
            return redirect(conn, '/vote')

        data = ''
        if data_list[0][3] == 'open' or data_list[0][3] == 'n_open':
            data += '<a href="/vote/close/' + num + '">(' + await get_lang('close_vote') + ')</a>'
        else:
            data += '<a href="/vote/close/' + num + '">(' + await get_lang('re_open_vote') + ')</a>'
        
        curs.execute(db_change('select data from vote where id = ? and name = "end_date" and type = "option"'), [num])
        db_data = curs.fetchall()
        time_limit = ''
        if db_data:
            time_limit = db_data[0][0]

        data += '<h2>' + data_list[0][0] + '</h2>'
        data += '<b>' + data_list[0][1] + '</b><hr class="main_hr">' if data_list[0][1] != '' else ''
        data += '<span>~ ' + time_limit + '</span><hr class="main_hr">' if time_limit != '' else ''

        vote_data = re.findall(r'([^\n]+)', data_list[0][2].replace('\r', ''))
        for i in range(0, len(vote_data)):
            data += '<h2>' + vote_data[i] + '</h2>'
            data += '<ul>'

            curs.execute(db_change('select user from vote where id = ? and user != "" and data = ?'), [num, str(i)])
            data_list_2 = curs.fetchall()
            if data_list[0][3] == 'open' or data_list[0][3] == 'close':
                all_ip = await ip_pas([j[0] for j in data_list_2])
                for j in data_list_2:
                    data += '<li>' + all_ip[j[0]] + '</li>'

            data += '<li>' + await get_lang('result') + ' : ' + str(len(data_list_2)) + '</li>'
            data += '</ul>'

        return await render_template(
            await get_lang('result_vote'),
            data,
            '(' + num + ')',
            [['vote', await get_lang('return')]]
        )