Source code for honeybee_plus.utilcol

"""A collection of useful utilities for Honeybee"""
import uuid
import re


def random_name(shorten=True):
    """Generate a random name as a string using uuid.

    Args:
        shorten: If True the name will be the first to segment of uuid.
    """
    if shorten:
        return '-'.join(str(uuid.uuid4()).split('-')[:2])
    else:
        return str(uuid.uuid4())


def check_name(name):
    """Check if a name is a valid honeybee name.

    A valid name can only have alphabet, digits, - and _.
    """
    name = name.encode('utf-8')
    try:
        match = re.match(b"^[.A-Za-z0-9_-]*$", name)
    except TypeError:
        match = re.match(r"^[.A-Za-z0-9_-]*$", name)
    
    if match:
        return True
    else:
        raise ValueError(
            'Invalid input name: ({}).'
            ' Name can only contain letters, numbers,'
            ' dots, underscores and dashes.'.format(name)
        )


if __name__ == '__main__':
    check_name('should_be_fine')
#     check_name('also-fine')
    check_name('this.is.also.fine.1234')
#     check_name('not good')