mirror of
https://github.com/kevin-DL/ShortMe-URL-Shortener.git
synced 2026-01-11 11:04:28 +00:00
113 lines
3.1 KiB
Python
113 lines
3.1 KiB
Python
# ------- standard library imports -------
|
|
import requests
|
|
|
|
# ------- 3rd party imports -------
|
|
from flask_restful import Resource, reqparse
|
|
|
|
# ------- local imports -------
|
|
from app.server.db.extensions import db
|
|
from app.server.api.api_auth import auth
|
|
from app.server.db.models import Url, AuthToken
|
|
|
|
shorten_parser = reqparse.RequestParser()
|
|
total_clicks_parser = reqparse.RequestParser()
|
|
|
|
shorten_parser.add_argument('url', type=str, help='URL parameter is missing', required=True)
|
|
total_clicks_parser.add_argument('url', type=str, help='Short URL is missing', required=True)
|
|
|
|
|
|
class Shorten(Resource):
|
|
"""
|
|
Return a short URL from a given URL
|
|
URL: /api/shorten
|
|
METHOD: POST
|
|
PARAMS: url: long URL
|
|
HEADERS: Authorization: (Bearer <key>)
|
|
RETURN: dictionary with the short URL
|
|
"""
|
|
|
|
decorators = [auth.login_required]
|
|
|
|
@staticmethod
|
|
def post():
|
|
args = shorten_parser.parse_args()
|
|
url = args['url']
|
|
original_url = url if url.startswith('http') else ('http://' + url)
|
|
|
|
try:
|
|
res = requests.get(original_url)
|
|
|
|
if res.status_code == 200:
|
|
url = Url(original_url=original_url)
|
|
|
|
db.session.add(url)
|
|
db.session.commit()
|
|
|
|
return dict(
|
|
short_url=url.short_url,
|
|
original_url=url.original_url,
|
|
success=True
|
|
), 200
|
|
|
|
else:
|
|
"""in case of page_not_found response from the URL given"""
|
|
return dict(
|
|
success=False,
|
|
message='could not shorten this URL (page_not_found)'
|
|
), 404
|
|
|
|
except (requests.exceptions.MissingSchema, requests.exceptions.ConnectionError, requests.exceptions.InvalidURL):
|
|
return dict(
|
|
success=False,
|
|
message='could not shorten this URL (page_not_found)'
|
|
), 404
|
|
|
|
|
|
class TotalClicks(Resource):
|
|
"""
|
|
return the total clicks of a short URL
|
|
URL: /api/total_clicks
|
|
METHOD: GET
|
|
PARAMS: short url
|
|
RETURN: dictionary with the total short url visits.
|
|
"""
|
|
|
|
@staticmethod
|
|
def get():
|
|
args = total_clicks_parser.parse_args()
|
|
url = args['url'].split('/')[-1]
|
|
|
|
try:
|
|
url = Url.query.filter_by(short_url=url).first()
|
|
|
|
return dict(
|
|
total=url.visits,
|
|
short_url=url.short_url,
|
|
original_url=url.original_url,
|
|
success=True
|
|
), 200
|
|
|
|
except AttributeError:
|
|
return dict(
|
|
success=False,
|
|
message='could not find the URL (page_not_found)'
|
|
), 404
|
|
|
|
|
|
class GetToken(Resource):
|
|
"""
|
|
Return a unique API authorization token. Used only internaly by the web app.
|
|
URL: /api/get_token
|
|
METHOD: GET
|
|
RETURN: unique API authorization token
|
|
AuthToken: Required
|
|
"""
|
|
decorators = [auth.login_required]
|
|
|
|
@staticmethod
|
|
def get():
|
|
token = AuthToken()
|
|
db.session.add(token)
|
|
db.session.commit()
|
|
return str(token)
|