import uuid

from bulksms.models import Bulk, Sms

def get_bulk_code(smscustomer):
    bulk_code = ''
    finish = False
    while not finish:
        bulk_code = 'BULK-' + str(uuid.uuid4())
        try:
            bulksms = Bulk.objects.get(bulk_code=bulk_code, customer=smscustomer)
        except Bulk.DoesNotExist:
            finish = True
    return bulk_code

def get_text_id(smscustomer):
    text_id = ''
    finish = False
    while not finish:
        text_id = 'SMS-' + str(uuid.uuid4())
        try:
            sms = Sms.objects.get(sms_id=text_id, customer=smscustomer)
        except Sms.DoesNotExist:
            finish = True
    return text_id

def get_bulk(bulk_code=None, smscustomer=None):
    try:
        bulksms = Bulk.objects.get(bulk_code=bulk_code, customer=smscustomer)
        return bulksms
    except Bulk.DoesNotExist:
        return None

def get_sms(text_id=None, smscustomer=None):
    try:
        sms = Sms.objects.get(sms_id=text_id, customer=smscustomer)
        return sms
    except Sms.DoesNotExist:
        return None