Files
digiserver/app.py
2025-01-13 15:36:39 +02:00

111 lines
3.9 KiB
Python

from flask import Flask, render_template, request, redirect, url_for
from flask_sqlalchemy import SQLAlchemy
app = Flask(__name__)
# Configurare baza de date SQLite
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///dashboard.db'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
db = SQLAlchemy(app)
# Modele pentru baza de date
class Player(db.Model):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(80), nullable=False)
hostname = db.Column(db.String(120), nullable=False)
ip = db.Column(db.String(15), nullable=False)
password = db.Column(db.String(120), nullable=False)
class Group(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(80), unique=True, nullable=False)
players = db.relationship('Player', secondary='group_players', backref='groups')
group_players = db.Table('group_players',
db.Column('group_id', db.Integer, db.ForeignKey('group.id'), primary_key=True),
db.Column('player_id', db.Integer, db.ForeignKey('player.id'), primary_key=True)
)
class Content(db.Model):
id = db.Column(db.Integer, primary_key=True)
file_name = db.Column(db.String(120), nullable=False)
duration = db.Column(db.Integer, nullable=False)
player_id = db.Column(db.Integer, db.ForeignKey('player.id'), nullable=True)
group_id = db.Column(db.Integer, db.ForeignKey('group.id'), nullable=True)
@app.before_first_request
def create_tables():
db.create_all()
@app.route('/')
def dashboard():
players = Player.query.all()
groups = Group.query.all()
return render_template('dashboard.html', players=players, groups=groups)
@app.route('/add_player', methods=['GET', 'POST'])
def add_player():
if request.method == 'POST':
username = request.form['username']
hostname = request.form['hostname']
ip = request.form['ip']
password = request.form['password']
new_player = Player(username=username, hostname=hostname, ip=ip, password=password)
db.session.add(new_player)
db.session.commit()
return redirect(url_for('dashboard'))
return render_template('add_player.html')
@app.route('/add_group', methods=['GET', 'POST'])
def add_group():
if request.method == 'POST':
group_name = request.form['group_name']
selected_players = request.form.getlist('players')
new_group = Group(name=group_name)
for player_id in selected_players:
player = Player.query.get(int(player_id))
new_group.players.append(player)
db.session.add(new_group)
db.session.commit()
return redirect(url_for('dashboard'))
players = Player.query.all()
return render_template('add_group.html', players=players)
@app.route('/upload_content', methods=['GET', 'POST'])
def upload_content():
if request.method == 'POST':
target_type = request.form['target_type']
target_id = request.form['target_id']
files = request.files.getlist('files')
duration = int(request.form['duration'])
for file in files:
if target_type == 'player':
new_content = Content(file_name=file.filename, duration=duration, player_id=int(target_id))
elif target_type == 'group':
new_content = Content(file_name=file.filename, duration=duration, group_id=int(target_id))
db.session.add(new_content)
db.session.commit()
return redirect(url_for('dashboard'))
players = Player.query.all()
groups = Group.query.all()
return render_template('upload_content.html', players=players, groups=groups)
@app.route('/player/<int:player_id>')
def player_page(player_id):
player = Player.query.get_or_404(player_id)
content = Content.query.filter_by(player_id=player_id).all()
return render_template('player_page.html', player=player, content=content)
if __name__ == '__main__':
app.run(debug=True)