Skip to content

Commit

Permalink
using furl to clean url
Browse files Browse the repository at this point in the history
  • Loading branch information
iambibhas committed Aug 25, 2020
1 parent 73f9bf4 commit 2d20465
Showing 1 changed file with 5 additions and 8 deletions.
13 changes: 5 additions & 8 deletions baseframe/filters.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
# -*- coding: utf-8 -*-

from __future__ import unicode_literals
from six.moves.urllib.parse import urlparse, urlunparse
import six

from datetime import datetime, timedelta
Expand All @@ -10,6 +9,7 @@
from flask import Markup, request

from babel.dates import format_date, format_datetime, format_time
from furl import furl
from pytz import UTC

from coaster.gfm import markdown
Expand Down Expand Up @@ -289,10 +289,7 @@ def timestamp_filter(value):

@baseframe.app_template_filter('cleanurl')
def cleanurl_filter(url):
parsed_url = urlparse(url)
unparsed_url = urlunparse(('', parsed_url.netloc, parsed_url.path, '', '', ''))
if unparsed_url.startswith('//'):
unparsed_url = unparsed_url.lstrip('//')
if unparsed_url.endswith('/'):
unparsed_url = unparsed_url.rstrip('/')
return unparsed_url
if not isinstance(url, furl):
url = furl(url)
url.path.normalize()
return furl().set(netloc=url.netloc, path=url.path).url.lstrip('//').rstrip('/')

0 comments on commit 2d20465

Please sign in to comment.