python strip_tags 支持保留指定标签

#coding:utf-8



import re



def strip_tags(string, allowed_tags=''):

  if allowed_tags != '':

    # Get a list of all allowed tag names.

    allowed_tags = allowed_tags.split(',')

    allowed_tags_pattern = ['</?'+allowed_tag+'[^>]*>' for allowed_tag in allowed_tags]

    all_tags = re.findall(r'<[^>]+>', string, re.I)

    not_allowed_tags = []

    tmp = 0

    for tag in all_tags:

        for pattern in allowed_tags_pattern:

            rs = re.match(pattern,tag)

            if rs:

                tmp += 1

            else:

                tmp += 0

        if not tmp:

            not_allowed_tags.append(tag)

        tmp = 0

    for not_allowed_tag in not_allowed_tags:

        string = re.sub(re.escape(not_allowed_tag), '',string)

    print not_allowed_tags

  else:

    # If no allowed tags, remove all.

    string = re.sub(r'<[^>]*?>', '', string)

 

  return string