99 lines
3.3 KiB
99 lines
3.3 KiB
from sqlalchemy import Column, DateTime, String, Integer, Boolean
|
|
from sqlalchemy import ForeignKey, func
|
|
from sqlalchemy.orm import relationship, backref
|
|
from sqlalchemy.ext.declarative import declarative_base
|
|
from ee.core.logging import Log
|
|
import sys
|
|
|
|
Base = declarative_base()
|
|
|
|
|
|
class SiteDB(Base):
|
|
__tablename__ = 'Site'
|
|
id = Column(Integer, primary_key=True)
|
|
sitename = Column(String, unique=True)
|
|
|
|
site_type = Column(String)
|
|
cache_type = Column(String)
|
|
site_path = Column(String)
|
|
|
|
# Use default=func.now() to set the default created time
|
|
# of a site to be the current time when a
|
|
# Site record was created
|
|
|
|
created_on = Column(DateTime, default=func.now())
|
|
site_enabled = Column(Boolean, unique=False, default=True, nullable=False)
|
|
is_ssl = Column(Boolean, unique=False, default=False)
|
|
storage_fs = Column(String)
|
|
storage_db = Column(String)
|
|
|
|
def __init__(self):
|
|
# from sqlalchemy import create_engine
|
|
# self.engine = create_engine('sqlite:///orm_in_detail.sqlite')
|
|
self.sitename = sitename
|
|
self.site_type = site_type
|
|
self.cache_type = cache_type
|
|
self.site_path = site_path
|
|
self.created_on = created_on
|
|
self.site_enabled = site_enabled
|
|
self.is_ssl = is_ssl
|
|
self.storage_fs = storage_fs
|
|
self.storage_db = storage_db
|
|
|
|
# if __name__ == "__main__":
|
|
#
|
|
# from sqlalchemy import create_engine
|
|
# engine = create_engine('sqlite:///orm_in_detail.sqlite')
|
|
# from sqlalchemy.orm import sessionmaker
|
|
# session = sessionmaker()
|
|
# session.configure(bind=engine)
|
|
# Base.metadata.create_all(engine)
|
|
# s = session()
|
|
# newRec = SiteDB(sitename='exa.in', site_type='wp', cache_type='basic',
|
|
# site_path='/var/www', site_enabled=True, is_ssl=False, storage_fs='ext4',
|
|
# storage_db='mysql')
|
|
# s.add(newRec)
|
|
# s.commit()
|
|
# s.flush()
|
|
|
|
|
|
def addNewSite(self, site, stype, cache, path,
|
|
enabled=True, ssl=False, fs='ext4', db='mysql'):
|
|
db_path = self.app.config.get('site', 'db_path')
|
|
try:
|
|
from sqlalchemy import create_engine
|
|
engine = create_engine(db_path)
|
|
from sqlalchemy.orm import sessionmaker
|
|
session = sessionmaker()
|
|
session.configure(bind=engine)
|
|
Base.metadata.create_all(engine)
|
|
s = session()
|
|
newRec = SiteDB(sitename=site, site_type=stype, cache_type=cache,
|
|
site_path=path, site_enabled=enabled, is_ssl=ssl,
|
|
storage_fs=fs, storage_db=db)
|
|
s.add(newRec)
|
|
s.commit()
|
|
s.flush()
|
|
except Exception as e:
|
|
Log.error(self, "Unable to add site to database : {0}"
|
|
.format(e))
|
|
sys.exit(1)
|
|
|
|
|
|
def getSiteInfo(self, site):
|
|
db_path = self.app.config.get('site', 'db_path')
|
|
try:
|
|
from sqlalchemy import create_engine
|
|
engine = create_engine(db_path)
|
|
from sqlalchemy.orm import sessionmaker
|
|
session = sessionmaker()
|
|
session.configure(bind=engine)
|
|
Base.metadata.create_all(engine)
|
|
s = session()
|
|
q = s.query(SiteDB).filter_by(sitename=site).first()
|
|
s.flush()
|
|
return q
|
|
except Exception as e:
|
|
Log.error(self, "Unable to add site to database : {0}"
|
|
.format(e))
|
|
sys.exit(1)
|
|
|