wirechan/app.py
2024-10-11 00:04:02 +05:00

56 lines
No EOL
1.8 KiB
Python

from flask import Flask, url_for, redirect, render_template, request
from pymongo import MongoClient
from datetime import datetime
from bson.objectid import ObjectId
app = Flask(__name__)
#initialize the database
client = MongoClient('localhost', 27017)
db = client.flask_db
posts_collection = db.posts_collection
#app routes
@app.route('/', methods=['GET'])
def index():
return render_template('index.html')
@app.route('/board/<board_name>', methods=['GET'])
def board(board_name):
links = [
{'name': 'b', 'display_name': '/b/ - random'},
{'name': 'g', 'display_name': '/g/ - technology'},
{'name': 'a', 'display_name': '/a/ - anime'},
{'name': 'v', 'display_name': '/v/ - video games'},
{'name': 'w', 'display_name': '/w/ - wallpapers'},
{'name': 'x', 'display_name': '/x/ - paranormal'},
{'name': 't', 'display_name': '/t/ - test'},
{'name': 's', 'display_name': '/s/ - soyjaks'},
{'name': 'pol', 'display_name': '/pol/ - politically incorrect'}
]
#global posts
#posts = posts_collection.find({'board_name': board_name}).sort('timestamp', -1)
display_name = next((link['display_name'] for link in links if link['name'] == board_name), board_name)
return render_template('board.html', title=board_name, header=display_name, links=links)
@app.route('/post', methods=['POST'])
def post():
board_name = request.form['board_name']
content = request.form['content']
image = request.files['image']
timestamp = datetime.now()
#insert the post into MongoDB
posts.insert_one({
'board_name': board_name,
'content': content,
'image': image.read(),
'timestamp': timestamp
})
return redirect(url_for('board', board_name=board_name))
if __name__ == '__main__':
app.run(debug=True)