Add models for charges

hitier
1 parent 25040dc3
Showing 3 changed files with 125 additions and 3 deletions Show diff stats
app/models.py
migrations/versions/6f7e51f380eb_.py
resources/db_config.py
@@ -7,9 +7,50 @@ 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(1000))
-    login = db.Column(db.String(1000), 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: {}, p: {}".format(self.id, self.name, self.email, self.login, self.password)
+        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))
+    secondname = db.Column(db.String(100))
+
+
+class Project(db.Model):
+    id = db.Column(db.Integer, primary_key=True)
+    name = db.Column(db.String(100), unique=True)
+
+
+class Service(db.Model):
+    id = db.Column(db.Integer, primary_key=True)
+    name = db.Column(db.String(100), unique=True)
+
+
+class Function(db.Model):
+    id = db.Column(db.Integer, primary_key=True)
+    name = db.Column(db.String(100), unique=True)
+
+
+class Capacity(db.Model):
+    id = db.Column(db.Integer, primary_key=True)
+    name = db.Column(db.String(100), unique=True)
+
+
+class Period(db.Model):
+    id = db.Column(db.Integer, primary_key=True)
+    name = db.Column(db.String(100), unique=True)
+    num_months = db.Column(db.Integer)
+
+
+class Charge(db.Model):
+    id = db.Column(db.Integer, primary_key=True)
+    agent_id = db.Column(db.Integer, db.ForeignKey('agent.id'))
+    project_id = db.Column(db.Integer, db.ForeignKey('project.id'))
+    service_id = db.Column(db.Integer, db.ForeignKey('service.id'))
+    period_id = db.Column(db.Integer, db.ForeignKey('period.id'))
+    charge_rate = db.Column(db.Integer)
@@ -0,0 +1,54 @@
+"""empty message
+
+Revision ID: 6f7e51f380eb
+Revises: 20f2c14aa0b6
+Create Date: 2021-03-09 09:47:34.679802
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = '6f7e51f380eb'
+down_revision = '20f2c14aa0b6'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table('agent',
+    sa.Column('id', sa.Integer(), nullable=False),
+    sa.Column('name', sa.String(length=1000), nullable=True),
+    sa.PrimaryKeyConstraint('id'),
+    sa.UniqueConstraint('name')
+    )
+    op.create_table('fonction',
+    sa.Column('id', sa.Integer(), nullable=False),
+    sa.Column('name', sa.String(length=1000), nullable=True),
+    sa.PrimaryKeyConstraint('id'),
+    sa.UniqueConstraint('name')
+    )
+    op.create_table('project',
+    sa.Column('id', sa.Integer(), nullable=False),
+    sa.Column('name', sa.String(length=1000), nullable=True),
+    sa.PrimaryKeyConstraint('id'),
+    sa.UniqueConstraint('name')
+    )
+    op.create_table('service',
+    sa.Column('id', sa.Integer(), nullable=False),
+    sa.Column('name', sa.String(length=1000), nullable=True),
+    sa.PrimaryKeyConstraint('id'),
+    sa.UniqueConstraint('name')
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_table('service')
+    op.drop_table('project')
+    op.drop_table('fonction')
+    op.drop_table('agent')
+    # ### end Alembic commands ###
@@ -0,0 +1,27 @@
+POSTGRES = {
+    'user': 'postgres',
+    'pw': 'postgres',
+    'db': 'climso',
+    'host': '127.0.0.1',
+    'port': '5434',
+}
+postgres_uri = 'postgresql://%(user)s:%(pw)s@%(host)s:%(port)s/%(db)s' % POSTGRES
+
+SQLITE = {
+    'file': '/home/richard/tmp/climso.db'
+}
+sqlite_uri = 'sqlite:///%(file)s' % SQLITE
+
+database_uri = sqlite_uri
+database_uri = postgres_uri
+
+data_basedir = '/data/CLIMSO'
+
+MYSQL = {
+    'user': 'mysql',
+    'pw': 'mysql',
+    'db': 'pdc_dev',
+    'host': '127.0.0.1',
+    'port': '3306',
+}
+mysql_uri = 'mysql+pymysql://%(user)s:%(pw)s@%(host)s:%(port)s/%(db)s' % MYSQL