Add the original models and __init__.py files for hotfix

Signed-off-by: Sayan Chowdhury <sayanchowdhury@fedoraproject.org>

files/hotfix/autocloud/models.py

@@ -0,0 +1,121 @@
+# -*- coding: utf-8 -*-
+
+import datetime
+
+from sqlalchemy import Column, Integer, String, DateTime, Text
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker
+from sqlalchemy.orm import scoped_session
+from sqlalchemy_utils import ChoiceType
+
+import autocloud
+
+Base = declarative_base()
+
+
+class JobDetails(Base):
+    __tablename__ = 'job_details'
+
+    STATUS_TYPES = (
+        ('s', 'Success'),
+        ('f', 'Failed'),
+        ('a', 'Aborted'),
+        ('r', 'Running'),
+        ('q', 'Queued')
+    )
+
+    IMAGE_FAMILY_TYPES = (
+        ('b', 'Base'),
+        ('a', 'Atomic')
+    )
+
+    ARCH_TYPES = (
+        ('i386', 'i386'),
+        ('x86_64', 'x86_64')
+    )
+
+    id = Column(Integer, primary_key=True)
+    taskid = Column(String(255), nullable=False)
+    status = Column(ChoiceType(STATUS_TYPES))
+    family = Column(ChoiceType(IMAGE_FAMILY_TYPES))
+    arch = Column(ChoiceType(ARCH_TYPES))
+    release = Column(String(255))
+    output = Column(Text, nullable=False, default='')
+    created_on = Column(DateTime, default=datetime.datetime.utcnow)
+    last_updated = Column(DateTime, default=datetime.datetime.utcnow)
+    user = Column(String(255), nullable=False)
+
+
+class ComposeDetails(Base):
+    __tablename__ = 'compose_details'
+
+    STATUS_TYPES = (
+        ('c', 'Complete'),
+        ('q', 'Queued'),
+        ('r', 'Running'),
+    )
+    id = Column(Integer, primary_key=True)
+    date = Column(DateTime, nullable=False)
+    compose_id = Column(String(255), nullable=False, unique=True)
+    respin = Column(Integer, nullable=False)
+    type = Column(String(255), nullable=False)
+    passed = Column(Integer, nullable=True, default=0)
+    failed = Column(Integer, nullable=True, default=0)
+    status = Column(ChoiceType(STATUS_TYPES))
+    created_on = Column(DateTime, default=datetime.datetime.utcnow)
+    last_updated = Column(DateTime, default=datetime.datetime.utcnow)
+    location = Column(String(255), nullable=False)
+
+
+class ComposeJobDetails(Base):
+    __tablename__ = 'compose_job_details'
+
+    STATUS_TYPES = (
+        ('s', 'Success'),
+        ('f', 'Failed'),
+        ('a', 'Aborted'),
+        ('r', 'Running'),
+        ('q', 'Queued')
+    )
+
+    IMAGE_FAMILY_TYPES = (
+        ('b', u'Base'),
+        ('a', u'Atomic')
+    )
+
+    ARCH_TYPES = (
+        ('i386', 'i386'),
+        ('x86_64', 'x86_64')
+    )
+
+    id = Column(Integer, primary_key=True)
+    arch = Column(ChoiceType(ARCH_TYPES))
+    compose_id = Column(String(255), nullable=False)
+    created_on = Column(DateTime, default=datetime.datetime.utcnow)
+    family = Column(ChoiceType(IMAGE_FAMILY_TYPES))
+    image_url = Column(String(255), nullable=False)
+    last_updated = Column(DateTime, default=datetime.datetime.utcnow)
+    output = Column(Text, nullable=False, default='')
+    release = Column(String(255))
+    status = Column(ChoiceType(STATUS_TYPES))
+    subvariant = Column(String(255), nullable=False)
+    user = Column(String(255), nullable=False)
+    image_format = Column(String(255), nullable=False)
+    image_type = Column(String(255), nullable=False)
+    image_name = Column(String(255), nullable=False)
+
+
+def create_tables():
+    # Create an engine that stores data in the local directory
+    engine = create_engine(autocloud.SQLALCHEMY_URI)
+
+    # Create all tables in the engine. This is equivalent to "Create Table"
+    # statements in raw SQL.
+    Base.metadata.create_all(engine)
+
+
+def init_model():
+    engine = create_engine(autocloud.SQLALCHEMY_URI)
+    scopedsession = scoped_session(sessionmaker(bind=engine))
+    return scopedsession