@@ -168,15 +168,15 @@ def sanitize_token(self, token):
168168if token_type in (tokenTypes ["StartTag" ],tokenTypes ["EndTag" ],
169169tokenTypes ["EmptyTag" ]):
170170if token ["name" ]in self .allowed_elements :
171- return self .allowed_token (token )
171+ return self .allowed_token (token , token_type )
172172else :
173- return self .unallowed_token (token )
173+ return self .unallowed_token (token , token_type )
174174elif token_type == tokenTypes ["Comment" ]:
175175pass
176176else :
177177return token
178178
179- def allowed_token (self ,token ):
179+ def allowed_token (self ,token , token_type ):
180180if "data" in token :
181181attrs = dict ([(name ,val )for name ,val in
182182token ["data" ][::- 1 ]
@@ -206,7 +206,7 @@ def allowed_token(self, token):
206206token ["data" ]= [[name ,val ]for name ,val in list (attrs .items ())]
207207return token
208208
209- def unallowed_token (self ,token ):
209+ def unallowed_token (self ,token , token_type ):
210210if token_type == tokenTypes ["EndTag" ]:
211211token ["data" ]= "</%s>" % token ["name" ]
212212elif token ["data" ]: