import os
import sys
sys.path.append(os.path.join(os.path.dirname(__file__), '..'))

from connector.mysql import get_ar_mysql_conection
from sql import subaccount

print('START')

try:
    LIMIT = int(sys.argv[1])
except ValueError:
    print("Please provide a numeric value as an argument.")
    exit()

print('FETCH %s PROJECTS' % LIMIT)
ar_db = get_ar_mysql_conection()
cursor = ar_db.cursor()
cursor.execute(subaccount.SELECTSQL.format(limit=LIMIT))
projects = cursor.fetchall()
if projects:
    for project_data in projects:
        try:
            cursor.execute(subaccount.UPDATESQL.format(project_id=project_data['project_id']))
            ar_db.commit()
            print('UPDATE SUCCESS FOR project_id %s' % project_data['project_id'])
        except:
            ar_db.rollback()
            print('UPDATE FAILED FOR project_id %s' %project_data['project_id'])
    ar_db.close()

print('COMPLETE')