batman (working version kinda)

app/core/auth.py

@@ -0,0 +1,31 @@
+from flask_login import LoginManager, UserMixin
+from werkzeug.security import generate_password_hash, check_password_hash
+from .extensions import db
+from datetime import datetime
+
+login_manager = LoginManager()
+login_manager.login_view = 'auth.login'
+
+class User(UserMixin, db.Model):
+    id = db.Column(db.Integer, primary_key=True)
+    email = db.Column(db.String(120), unique=True, nullable=False)
+    password_hash = db.Column(db.String(128), nullable=False)
+    is_admin = db.Column(db.Boolean, default=False)
+    created_at = db.Column(db.DateTime, default=datetime.utcnow)
+    last_seen = db.Column(db.DateTime, default=datetime.utcnow)
+    
+    def set_password(self, password):
+        self.password_hash = generate_password_hash(password)
+
+    def check_password(self, password):
+        return check_password_hash(self.password_hash, password)
+
+    def get_id(self):
+        return str(self.id)
+
+    def __repr__(self):
+        return f'<User {self.email}>'
+
+@login_manager.user_loader
+def load_user(user_id):
+    return User.query.get(int(user_id))

app/core/extensions.py

@@ -0,0 +1,28 @@
+from flask_sqlalchemy import SQLAlchemy
+from flask_bcrypt import Bcrypt
+from flask_limiter import Limiter
+from flask_limiter.util import get_remote_address
+from flask_login import LoginManager
+from flask_wtf.csrf import CSRFProtect
+
+# Initialize extensions
+db = SQLAlchemy()
+bcrypt = Bcrypt()
+login_manager = LoginManager()
+login_manager.login_view = 'auth.login'
+csrf = CSRFProtect()
+
+# Initialize rate limiter with fallback storage
+try:
+    limiter = Limiter(
+        key_func=get_remote_address,
+        default_limits=["200 per day", "50 per hour"],
+        storage_uri="memory://"  # Use memory storage for development
+    )
+except Exception as e:
+    print(f"Error initializing rate limiter: {e}")
+    # Fallback limiter with very basic functionality
+    limiter = Limiter(
+        key_func=get_remote_address,
+        default_limits=["200 per day", "50 per hour"]
+    )

app/core/models.py

@@ -0,0 +1,77 @@
+from .extensions import db
+import json
+from datetime import datetime
+import ipaddress
+
+class Subnet(db.Model):
+    id = db.Column(db.Integer, primary_key=True)
+    cidr = db.Column(db.String(18), unique=True)  # Format: "192.168.1.0/24"
+    location = db.Column(db.String(80))
+    auto_scan = db.Column(db.Boolean, default=False)
+    created_at = db.Column(db.DateTime, default=datetime.utcnow)
+    
+    @property
+    def network(self):
+        return ipaddress.ip_network(self.cidr)
+    
+    @property
+    def num_addresses(self):
+        return self.network.num_addresses
+    
+    @property
+    def used_ips(self):
+        # Count servers in this subnet
+        return Server.query.filter_by(subnet_id=self.id).count()
+    
+    def __repr__(self):
+        return f'<Subnet {self.cidr}>'
+
+
+class Server(db.Model):
+    id = db.Column(db.Integer, primary_key=True)
+    hostname = db.Column(db.String(80), unique=True)
+    ip_address = db.Column(db.String(15), unique=True)
+    subnet_id = db.Column(db.Integer, db.ForeignKey('subnet.id'))
+    subnet = db.relationship('Subnet', backref=db.backref('servers', lazy=True))
+    documentation = db.Column(db.Text)
+    created_at = db.Column(db.DateTime, default=datetime.utcnow)
+    
+    # Store ports as JSON in the database
+    _ports = db.Column(db.Text, default='[]')
+    
+    @property
+    def ports(self):
+        return json.loads(self._ports) if self._ports else []
+    
+    @ports.setter
+    def ports(self, value):
+        self._ports = json.dumps(value) if value else '[]'
+    
+    def get_open_ports(self):
+        return self.ports
+    
+    def __repr__(self):
+        return f'<Server {self.hostname}>'
+
+
+class App(db.Model):
+    id = db.Column(db.Integer, primary_key=True)
+    name = db.Column(db.String(80))
+    server_id = db.Column(db.Integer, db.ForeignKey('server.id'))
+    server = db.relationship('Server', backref=db.backref('apps', lazy=True))
+    documentation = db.Column(db.Text)
+    created_at = db.Column(db.DateTime, default=datetime.utcnow)
+    
+    # Store ports as JSON in the database
+    _ports = db.Column(db.Text, default='[]')
+    
+    @property
+    def ports(self):
+        return json.loads(self._ports) if self._ports else []
+    
+    @ports.setter
+    def ports(self, value):
+        self._ports = json.dumps(value) if value else '[]'
+    
+    def __repr__(self):
+        return f'<App {self.name}>'