Move User model to auth module

hitier
1 parent 775de511
Showing 5 changed files with 20 additions and 17 deletions Show diff stats
app/__init__.py
app/auth/models.py
app/auth/routes.py
app/commands/commands.py
app/models.py
@@ -8,7 +8,8 @@ import logging
 from logging.handlers import RotatingFileHandler
 from flask.logging import default_handler
-from app.models import db, User
+from app.models import db
+from app.auth.models import User
 from dotenv import load_dotenv
@@ -0,0 +1,14 @@
+from flask_login import UserMixin
+from app.models import db
+
+
+class User(UserMixin, db.Model):
+    id = db.Column(db.Integer, primary_key=True)  # primary keys are required by SQLAlchemy
+    email = db.Column(db.String(100), unique=True)
+    name = db.Column(db.String(100))
+    login = db.Column(db.String(100), unique=True)
+    password = db.Column(db.String(100))
+
+    def __repr__(self):
+        return "i: {}, n: {}, e: {}, l: {}".format(self.id, self.name, self.email, self.login)
+
 from flask import render_template, request, redirect, url_for, flash
 from flask_login import login_user, logout_user
-from app.models import User
+from .models import User
 from . import bp
@@ -9,7 +9,8 @@ from sqlalchemy.ext.automap import automap_base
 from sqlalchemy.orm import Session
 from sqlalchemy import create_engine
-from app.models import db, User, Agent, Service, Project, Capacity, Period, Charge
+from app.models import db, Agent, Service, Project, Capacity, Period, Charge
+from app.auth.models import User
 from . import bp
@@ -32,7 +33,6 @@ def feed_from_lesia():
         print("Please, configure the mysql database (see db_config.py)")
         sys.exit(-1)
-
     # mapped classes are now created with names by default
     # matching that of the table name.
     LesiaAgent = Base.classes.agent
@@ -96,7 +96,7 @@ def feed_random_charges(agent):
             .filter(Charge.agent_id == agent_id,
                     Charge.period_id == period_id
                     ).scalar()
-        if total_charge is not None and (total_charge + percent)>= 100:
+        if total_charge is not None and (total_charge + percent) >= 100:
             print("Skipping agent {} for period {}".format(agent_id, period_id))
             continue
         charge = Charge(agent_id=agent_id,
-from flask_login import UserMixin
 from flask_sqlalchemy import SQLAlchemy
 db = SQLAlchemy()
-class User(UserMixin, db.Model):
-    id = db.Column(db.Integer, primary_key=True)  # primary keys are required by SQLAlchemy
-    email = db.Column(db.String(100), unique=True)
-    name = db.Column(db.String(100))
-    login = db.Column(db.String(100), unique=True)
-    password = db.Column(db.String(100))
-
-    def __repr__(self):
-        return "i: {}, n: {}, e: {}, l: {}".format(self.id, self.name, self.email, self.login)
-
-
 class Agent(db.Model):
     id = db.Column(db.Integer, primary_key=True)
     firstname = db.Column(db.String(100))