import MySQLdb
import math
import random

def Create(phenny, enemyStats, playerStats, parameters):
    db = MySQLdb.connect(host="localhost", user=phenny.config.mysql_username, passwd=phenny.config.mysql_password,db=parameters['LiveDB'])
    c = db.cursor(MySQLdb.cursors.DictCursor)

    # Update player state for combat
    c.execute("UPDATE `Players` SET `State`='Combat' WHERE `Nick`=%s;", (parameters['Nick']));db.commit();

    if playerStats['ChallengeRating'] == int(math.ceil(playerStats['ChallengeRating'] / 4) * 4):
        # Get total unluck enemy weighting
        c.execute("SELECT SUM(`EnemyWeight`) AS `EnemyWeight` FROM `GenEnemy` WHERE `Locked`=0 AND `Class`='Boss'")
        intTotalWeight = random.randint(1, int(c.fetchone()['EnemyWeight']))

        #Get Boss
        c.execute("SELECT `EnemyColor`, `EnemyName`, `Season`, `Class`, `Unlock`, `HP`, `Locked`, `EnemyWeight`, `ChallengeRating`, `GoldCarried`, `Strength`, `Dexterity`, `Constitution`, `Intelligence`, `Dodge`, `Crit` "+
                  "FROM `GenEnemy` WHERE `Locked`=0 AND `Class`='Boss'")
    else:
        # Get total unluck enemy weighting
        c.execute("SELECT SUM(`EnemyWeight`) AS `EnemyWeight` FROM `GenEnemy` WHERE `Locked`=0 AND `Class`='Rare' OR `Locked`=0 AND `Class`='Minor'")
        intTotalWeight = random.randint(1, int(c.fetchone()['EnemyWeight']))

        #Get enemy
        c.execute("SELECT `EnemyColor`, `EnemyName`, `Season`, `Class`, `Unlock`, `HP`, `Locked`, `EnemyWeight`, `ChallengeRating`, `GoldCarried`, `Strength`, `Dexterity`, `Constitution`, `Intelligence`, `Dodge`, `Crit` "+
                  "FROM `GenEnemy` WHERE `Locked`=0 AND `Class`='Rare' OR "+
                                        "`Locked`=0 AND `Class`='Minor'")
    enemyStats = c.fetchone()

    while True:
        intTotalWeight = intTotalWeight - enemyStats['EnemyWeight']
        if intTotalWeight <= 0:
            break
        else:
            enemyStats = c.fetchone()

    #Apply challenge rating modifier to enemy
    intChallengeModifier = 2
    enemyStats['HP'] = int(math.ceil(enemyStats['HP'] * playerStats['ChallengeRating']/intChallengeModifier))
    enemyStats['GoldCarried'] = int(math.ceil(enemyStats['GoldCarried'] * playerStats['ChallengeRating']/intChallengeModifier))
    enemyStats['Strength'] = int(math.ceil(enemyStats['Strength'] * playerStats['ChallengeRating']/intChallengeModifier))
    enemyStats['Dexterity'] = int(math.ceil(enemyStats['Dexterity'] * playerStats['ChallengeRating']/intChallengeModifier))
    enemyStats['Constitution'] = int(math.ceil(enemyStats['Constitution'] * playerStats['ChallengeRating']/intChallengeModifier))
    enemyStats['Intelligence'] = int(math.ceil(enemyStats['Intelligence'] * playerStats['ChallengeRating']/intChallengeModifier))

    #Create entity in database
    c.execute("INSERT INTO CombatEnemy(`EnemyColor`,`EnemyName`,`Season`,`Class`,`Nick`,`HP`,`Strength`,`Dexterity`,`Constitution`,`Intelligence`,`GoldCarried`,`Unlock`) VALUES ('"+
    str(enemyStats['EnemyColor'])+"','"+str(enemyStats['EnemyName'])+"','"+str(enemyStats['Season'])+"','"+str(enemyStats['Class'])+"','"+parameters['Nick']+"','"+str(enemyStats['HP'])+"','"+str(enemyStats['Strength'])+"','"+str(enemyStats['Dexterity'])+"','"+str(enemyStats['Constitution'])+"','"+str(enemyStats['Intelligence'])+"','"+str(enemyStats['GoldCarried'])+"','"+str(enemyStats['Unlock'])+"')");db.commit();

    #Cleanup
    db.close()
    return enemyStats

def GetStats(phenny, parameters, enemyStats):
    db = MySQLdb.connect(host="localhost", user=phenny.config.mysql_username, passwd=phenny.config.mysql_password,db=parameters['LiveDB'])
    c = db.cursor(MySQLdb.cursors.DictCursor)
    c.execute("SELECT `EnemyColor`,`EnemyName`, `Class`, `Season`, `Unlock`,`HP`,`GoldCarried`,`Strength`,`Dexterity`,`Constitution`,`Intelligence`,`GoldCarried`, `Dodge`, `Crit` FROM `CombatEnemy` WHERE `Nick`=%s",parameters['Nick'])
    sqlEnemy = c.fetchone()
    enemyStats['EnemyColor'] = sqlEnemy['EnemyColor']
    enemyStats['EnemyName'] = sqlEnemy['EnemyName']
    enemyStats['Class'] = sqlEnemy['Class']
    enemyStats['Season'] = sqlEnemy['Season']
    enemyStats['Unlock'] = sqlEnemy['Unlock']
    enemyStats['HP'] = sqlEnemy['HP']
    enemyStats['GoldCarried'] = sqlEnemy['GoldCarried']
    enemyStats['Strength'] = sqlEnemy['Strength']
    enemyStats['Dexterity'] = sqlEnemy['Dexterity']
    enemyStats['Constitution'] = sqlEnemy['Constitution']
    enemyStats['Intelligence'] = sqlEnemy['Intelligence']
    enemyStats['DodgePercentage'] = sqlEnemy['Dodge']
    enemyStats['CritPercentage'] = sqlEnemy['Crit']
    enemyStats['ArmorPercentage'] = 0
    enemyStats['UnitType'] = 'Enemy'
    enemyStats['BlockPercentage'] = 0
    db.close()

    return enemyStats