first commit

lib/system/model.py

@@ -0,0 +1,158 @@
+# -*- coding: utf-8 -*-
+# =========================================================
+# python
+import traceback
+
+# third-party
+import requests
+from flask import (
+    Blueprint,
+    request,
+    Response,
+    send_file,
+    render_template,
+    redirect,
+    jsonify,
+)
+
+# gommi 공용
+from framework.logger import get_logger
+from framework import app, db, scheduler
+from framework.util import Util
+
+# 패키지
+
+# 로그
+package_name = __name__.split(".")[0]
+logger = get_logger(package_name)
+# ==============================================================
+
+
+class ModelSetting(db.Model):
+    __tablename__ = "system_setting"
+    id = db.Column(db.Integer, primary_key=True)
+    key = db.Column(db.String(100), unique=True, nullable=False)
+    value = db.Column(db.String(100), nullable=False)
+
+    def __init__(self, key, value):
+        self.key = key
+        self.value = value
+
+    def __repr__(self):
+        return "<SystemSetting(id:%s, key:%s, value:%s)>" % (
+            self.id,
+            self.key,
+            self.value,
+        )
+
+    def as_dict(self):
+        return {x.name: getattr(self, x.name) for x in self.__table__.columns}
+
+    @staticmethod
+    def get(key):
+        try:
+            ret = db.session.query(ModelSetting).filter_by(key=key).first()
+            if ret is not None:
+                return ret.value.strip()
+            else:
+                return ""
+        except Exception as exception:
+            logger.error("Exception:%s %s", exception, key)
+            logger.error(traceback.format_exc())
+
+    @staticmethod
+    def get_int(key):
+        try:
+            return int(ModelSetting.get(key))
+        except Exception as exception:
+            logger.error("Exception:%s %s", exception, key)
+            logger.error(traceback.format_exc())
+
+    @staticmethod
+    def get_bool(key):
+        try:
+            return ModelSetting.get(key) == "True"
+        except Exception as exception:
+            logger.error("Exception:%s %s", exception, key)
+            logger.error(traceback.format_exc())
+
+    @staticmethod
+    def set(key, value):
+        try:
+            logger.debug(key)
+            item = (
+                db.session.query(ModelSetting)
+                .filter_by(key=key)
+                .with_for_update()
+                .first()
+            )
+            if item is not None:
+                item.value = value.strip() if value is not None else value
+                db.session.commit()
+            else:
+                db.session.add(ModelSetting(key, value.strip()))
+        except Exception as exception:
+            logger.error("Exception:%s %s", exception, key)
+            logger.error(traceback.format_exc())
+
+    @staticmethod
+    def to_dict():
+        try:
+            from framework.util import Util
+
+            arg = Util.db_list_to_dict(db.session.query(ModelSetting).all())
+            arg["package_name"] = package_name
+            return arg
+        except Exception as exception:
+            logger.error("Exception:%s", exception)
+            logger.error(traceback.format_exc())
+
+    @staticmethod
+    def setting_save(req):
+        try:
+            for key, value in req.form.items():
+                if key in ["scheduler", "is_running"]:
+                    continue
+                if key.startswith("tmp_"):
+                    continue
+                logger.debug("Key:%s Value:%s", key, value)
+                entity = (
+                    db.session.query(ModelSetting)
+                    .filter_by(key=key)
+                    .with_for_update()
+                    .first()
+                )
+                entity.value = value
+            db.session.commit()
+            return True
+        except Exception as exception:
+            logger.error("Exception:%s", exception)
+            logger.error(traceback.format_exc())
+            logger.debug("Error Key:%s Value:%s", key, value)
+            return False
+
+    @staticmethod
+    def get_list(key):
+        try:
+            value = ModelSetting.get(key)
+            values = [
+                x.strip().replace(" ", "").strip()
+                for x in value.replace("\n", "|").split("|")
+            ]
+            values = Util.get_list_except_empty(values)
+            return values
+        except Exception as exception:
+            logger.error("Exception:%s", exception)
+            logger.error(traceback.format_exc())
+            logger.error("Error Key:%s Value:%s", key, value)
+
+    @staticmethod
+    def make_apikey(url):
+        url = url.format(ddns=ModelSetting.get("ddns"))
+        if ModelSetting.get_bool("auth_use_apikey"):
+            if url.find("?") == -1:
+                url += "?"
+            else:
+                url += "&"
+            url += "apikey=%s" % ModelSetting.get("auth_apikey")
+        return url