masschallenge/django-accelerator

import re

from .utils import (
    remove_leading_slashes,
    remove_trailing_slashes,
    remove_trailing_and_leading_whitespace,
    remove_leading_hashtag_on_valid_twitter_handles,
    remove_twitter_url_prefix_from_handles,
    remove_not_available_abbreviation_from_twitter_handles,
    turn_handles_with_incomplete_handles_to_empty_string
)


def clean_entrepreneur_profile_twitter_handles(EntrepreneurProfile):
    remove_twitter_url_prefix_from_handles(EntrepreneurProfile)
    turn_handles_with_incomplete_handles_to_empty_string(EntrepreneurProfile)
    remove_not_available_abbreviation_from_twitter_handles(
        EntrepreneurProfile)
    remove_leading_hashtag_on_valid_twitter_handles(EntrepreneurProfile)

    for ent in EntrepreneurProfile.objects.exclude(twitter_handle=""):
        twitter_handle = ent.twitter_handle
        match = re.match(r'^@?(\w){1,15}$', twitter_handle)
        if not match:
            remove_trailing_and_leading_whitespace(ent)
            remove_leading_slashes(ent)
            remove_trailing_slashes(ent)