1
0
Fork 0
forked from mia/0x0

add BEACON dump with start index for urlte.am

fixes #1
This commit is contained in:
Martin Herkt 2017-01-01 20:27:31 +01:00
parent e974834422
commit 6bc7690711
No known key found for this signature in database
GPG key ID: C24B9CD04DC6AE7F

View file

@ -1,7 +1,7 @@
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
from flask import Flask, abort, escape, make_response, redirect, request, send_from_directory, url_for
from flask import Flask, abort, escape, make_response, redirect, request, send_from_directory, url_for, Response
from flask_sqlalchemy import SQLAlchemy
from flask_script import Manager
from flask_migrate import Migrate, MigrateCommand
@ -243,6 +243,24 @@ def get(path):
abort(404)
@app.route("/dump_urls/")
@app.route("/dump_urls/<int:start>")
def dump_urls(start=0):
meta = "#FORMAT: BEACON\n#PREFIX: {}/\n\n".format(fhost_url("https"))
def gen():
yield meta
for url in URL.query.order_by(URL.id.asc()).offset(start):
if url.url.startswith("http") or url.url.startswith("https"):
bar = "|"
else:
bar = "||"
yield url.getname() + bar + url.url + "\n"
return Response(gen(), mimetype="text/plain")
@app.route("/", methods=["GET", "POST"])
def fhost():
if request.method == "POST":