from otree.api import Currency as c, currency_range
from ._builtin import Page, WaitPage
from .models import Constants
import sqlite3
import os
from datetime import datetime, timedelta, date

class MyPage(Page):
    
    def vars_for_template(self):
        
        conn = sqlite3.connect('_static/iq_results/iq_data.db')
        c = conn.cursor()

        unique_id = str(self.player.participant.label)
        c.execute('SELECT * FROM results WHERE ID=?', (unique_id,) )
        
        try:
            self.player.correct_flag = 1
            self.player.uniqueID = unique_id
            row = c.fetchone()
            self.player.TOT_EARN = row[1]
            self.player.IQ = row[2]
            self.player.TASK = row[3]
            self.player.BOX = row[4]
            self.player.RANK = row[5]
            self.player.S = row[6]
            self.player.NUM = row[7]
            self.player.MONEY_TASK = row[8]
            self.player.TEST_DATE = row[9]
            
        except Exception as e:
            self.player.uniqueID = unique_id
            self.player.correct_flag = 0
            # self.player.TEST_DATE = str(e)
        
        test_date = datetime.strptime(self.player.TEST_DATE , "%d/%m/%Y").date()
            
        if (( date.today() - test_date).days >= 7):
            self.player.date_flag = 1
        else:
            self.player.date_flag = 0
            self.player.correct_flag = 0
            self.player.NEW_DATE = str(test_date + timedelta(days=7))
            

page_sequence = [
    MyPage
]