Added View Server Page & Discord API Handler
This commit is contained in:
28
main.py
28
main.py
@@ -22,19 +22,45 @@ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||
SOFTWARE.
|
||||
"""
|
||||
|
||||
from flask import Flask, render_template
|
||||
from flask import Flask, render_template, redirect, url_for
|
||||
|
||||
import humanfriendly
|
||||
|
||||
from PyJS import JSON
|
||||
from PyJS.modules import fs
|
||||
|
||||
from core import discordAPI
|
||||
from core.database import Servers
|
||||
from core.routes import route_for
|
||||
|
||||
app = Flask(__name__)
|
||||
|
||||
settings = JSON.parse(fs.createReadStream('settings.json'))
|
||||
settings["len"] = len
|
||||
settings["human"] = humanfriendly
|
||||
|
||||
@app.errorhandler(404)
|
||||
def _404Err(err):
|
||||
return redirect(url_for('index'))
|
||||
|
||||
@app.route("/")
|
||||
def index():
|
||||
return render_template('index.html', servers=Servers.get_all(),**settings)
|
||||
|
||||
@app.route(**route_for('view_server_default'), defaults={"serverID": None})
|
||||
@app.route(**route_for('view_server'))
|
||||
def view_server(serverID):
|
||||
if not serverID:
|
||||
redirect(url_for('index'))
|
||||
|
||||
server = Servers.get_one(_id=int(serverID))
|
||||
|
||||
_data = discordAPI.get_guild(serverID)
|
||||
|
||||
inv = discordAPI.get_invite(server['invite']).json()
|
||||
|
||||
_data["invite"]= f"https://discord.gg/{inv['code']}"
|
||||
|
||||
return render_template('view_server.html', server=server, discord=_data, **settings)
|
||||
|
||||
app.run(debug=True)
|
||||
Reference in New Issue
Block a user