Torma Kristóf
5e541640c1
All checks were successful
continuous-integration/drone/push Build is passing
66 lines
2.4 KiB
Python
66 lines
2.4 KiB
Python
#!/usr/bin/env python3
|
|
from enum import Enum
|
|
|
|
from sqlalchemy.dialects.postgresql import UUID
|
|
from sqlalchemy.orm import backref
|
|
from sqlalchemy.sql import func
|
|
from flask_bcrypt import generate_password_hash, check_password_hash
|
|
|
|
from db import db
|
|
|
|
"""
|
|
Database models
|
|
"""
|
|
|
|
__author__ = '@tormakris'
|
|
__copyright__ = "Copyright 2020, videON Team"
|
|
__module_name__ = "models"
|
|
__version__text__ = "1"
|
|
|
|
|
|
class VideonUser(db.Model):
|
|
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
|
|
|
|
name = db.Column(db.String, nullable=False, unique=True)
|
|
password = db.Column(db.String, nullable=False)
|
|
|
|
last_logon = db.Column(db.TIMESTAMP, nullable=False, server_default=func.now())
|
|
|
|
timestamp = db.Column(db.TIMESTAMP, nullable=False, server_default=func.now())
|
|
|
|
def hash_password(self):
|
|
self.password = generate_password_hash(self.password).decode('utf8')
|
|
|
|
def check_password(self, password):
|
|
return check_password_hash(self.password, password)
|
|
|
|
|
|
class StreamResourceTypeEnum(Enum):
|
|
INGEST = 1
|
|
ENCODE = 2
|
|
RESTREAM = 3
|
|
|
|
|
|
class StreamResource(db.Model):
|
|
id = db.Column(UUID(as_uuid=True), primary_key=True, unique=True, nullable=False,
|
|
server_default=db.text("uuid_generate_v4()"))
|
|
resource_type = db.Column(db.Enum(StreamResourceTypeEnum), nullable=False)
|
|
url = db.Column(db.String, nullable=False)
|
|
bitrate = db.Column(db.Integer, nullable=True)
|
|
width = db.Column(db.Integer, nullable=True)
|
|
height = db.Column(db.Integer, nullable=True)
|
|
parent_id = db.Column(UUID(as_uuid=True), db.ForeignKey('stream_resource.id'), nullable=True)
|
|
children = db.relationship('StreamResource', lazy=False)
|
|
owner_id = db.Column(db.Integer, db.ForeignKey('videon_user.id'), nullable=False)
|
|
stream_key = db.Column(UUID(as_uuid=True), unique=True, nullable=False)
|
|
owner = db.relationship('VideonUser', backref=backref("stream_resource", cascade="all, delete"), lazy=False)
|
|
x = db.Column(db.Integer, nullable=False)
|
|
y = db.Column(db.Integer, nullable=False)
|
|
|
|
|
|
class OutputUrls(db.Model):
|
|
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
|
|
output_url = db.Column(db.String, nullable=False)
|
|
streamresource_id = db.Column(UUID(as_uuid=True), db.ForeignKey('stream_resource.id'), nullable=False)
|
|
streamresource = db.relationship('StreamResource', backref=backref('output_urls', cascade="all, delete"), lazy=False)
|