"""Docstring."""

import os
import sys

from sqlalchemy import create_engine
from sqlalchemy.orm import Session

AR_CONN = None

def main():
    total = int(sys.argv[1])
    batch_size = 100
    while total > 0:
        results = claim_isrcs(batch_size)
        for x in results:
            print(x)
        total -= len(results)


def claim_isrcs(batch_size):
    query = 'CALL `sp_claim_isrcs_for_use`(:number_of_isrcs);'
    results = AR_CONN.execute(
        query,
        {
            'number_of_isrcs': batch_size
        }
    )
    return [x[0] for x in results.fetchall()]


if __name__ == '__main__':
    ar_user = os.getenv('AR_USER')
    ar_password = os.getenv('AR_PASSWORD')
    ar_host = os.getenv('AR_HOST')
    ar_engine = create_engine(f'mysql+pymysql://{ar_user}:{ar_password}@{ar_host}/art_relations')  # noqa:E501
    AR_CONN = Session(ar_engine)
    AR_CONN.execute('SELECT 1')
    try:
        main()
    finally:
        ar_engine.dispose()