Initial commit – AufmaßCreater v2.35

_aufmass_web/app/models/license.py

@@ -0,0 +1,59 @@
+from app.extensions import db
+from datetime import datetime
+import hashlib, secrets
+
+def _generate_uid(company_name):
+    raw = f"{company_name}-{secrets.token_hex(6)}"
+    return hashlib.sha256(raw.encode()).hexdigest()[:12]
+
+class License(db.Model):
+    __tablename__ = 'licenses'
+
+    id = db.Column(db.Integer, primary_key=True)
+    company_id = db.Column(db.Integer, db.ForeignKey('companies.id'), nullable=False)
+    uid = db.Column(db.String(64), unique=True, nullable=False)
+    max_mitarbeiter = db.Column(db.Integer, default=5)
+    max_module_slots = db.Column(db.Integer, default=5)
+    unlimited_users = db.Column(db.Boolean, default=False)
+    unlimited_modules = db.Column(db.Boolean, default=False)
+    aktiv = db.Column(db.Boolean, default=True)
+    erstellt_am = db.Column(db.DateTime, default=datetime.utcnow)
+
+    modules = db.relationship('LicenseModule', backref='license', lazy='dynamic', cascade='all, delete-orphan')
+
+    @property
+    def used_users(self):
+        from app.models.user import User
+        return User.query.filter_by(company_id=self.company_id).count()
+
+    @property
+    def used_module_slots(self):
+        from app.models.user_module import UserModulePermission
+        from app.models.user import User
+        return db.session.query(UserModulePermission.id).join(User, UserModulePermission.user_id==User.id)\
+            .filter(User.company_id==self.company_id, UserModulePermission.aktiv==True).count()
+
+    def user_slots_display(self):
+        if self.unlimited_users: return '\u221e'
+        return f'{self.used_users} / {self.max_mitarbeiter}'
+
+    def module_slots_display(self):
+        if self.unlimited_modules: return '\u221e'
+        return f'{self.used_module_slots} / {self.max_module_slots}'
+
+    @staticmethod
+    def generate_uid(company_name):
+        return _generate_uid(company_name)
+
+    def __repr__(self):
+        return f'<License {self.uid}>'
+
+class LicenseModule(db.Model):
+    __tablename__ = 'license_modules'
+
+    id = db.Column(db.Integer, primary_key=True)
+    license_id = db.Column(db.Integer, db.ForeignKey('licenses.id'), nullable=False)
+    module_id = db.Column(db.Integer, db.ForeignKey('modules.id'), nullable=False)
+    aktiv = db.Column(db.Boolean, default=True)
+
+    module = db.relationship('Module', backref='license_assignments')