jaseci 1.4.0.9__py3-none-any.whl → 1.4.0.11__py3-none-any.whl

jaseci/jsorc_settings.py

@@ -0,0 +1,184 @@
+import os
+import yaml
+
+
+def load_default_yaml(file):
+    manifest = {}
+    with open(
+        f"{os.path.dirname(os.path.abspath(__file__))}/manifests/{file}.yaml", "r"
+    ) as stream:
+        try:
+            for conf in yaml.safe_load_all(stream):
+                kind = conf["kind"]
+                if not manifest.get(kind):
+                    manifest[kind] = []
+                manifest[kind].append(conf)
+        except yaml.YAMLError as exc:
+            print(exc)
+
+    return manifest
+
+
+class JsOrcSettings:
+    ###############################################################################################################
+    # ------------------------------------------------- DEFAULT ------------------------------------------------- #
+    ###############################################################################################################
+
+    DEFAULT_CONFIG = {"enabled": False, "quiet": False, "automated": False}
+
+    DEFAULT_MANIFEST = {}
+
+    UNSAFE_PARAPHRASE = "I know what I'm doing!"
+    UNSAFE_KINDS = ["PersistentVolumeClaim"]
+
+    ###############################################################################################################
+    # -------------------------------------------------- JSORC -------------------------------------------------- #
+    ###############################################################################################################
+
+    JSORC_CONFIG = {
+        "backoff_interval": 10,
+        "pre_loaded_services": ["kube", "redis", "prome", "mail", "task", "elastic"],
+    }
+
+    ###############################################################################################################
+    # -------------------------------------------------- KUBE --------------------------------------------------- #
+    ###############################################################################################################
+
+    KUBE_CONFIG = {
+        "enabled": bool(os.getenv("KUBE_NAMESPACE")),
+        "quiet": True,
+        "automated": False,
+        "namespace": os.getenv("KUBE_NAMESPACE", "default"),
+        "in_cluster": True,
+        "config": None,
+    }
+
+    ###############################################################################################################
+    # -------------------------------------------------- REDIS -------------------------------------------------- #
+    ###############################################################################################################
+
+    REDIS_CONFIG = {
+        "enabled": True,
+        "quiet": True,
+        "automated": True,
+        "host": os.getenv("REDIS_HOST", "localhost"),
+        "port": os.getenv("REDIS_PORT", "6379"),
+        "db": os.getenv("REDIS_DB", "1"),
+    }
+
+    REDIS_MANIFEST = load_default_yaml("redis")
+
+    ###############################################################################################################
+    # -------------------------------------------------- TASK --------------------------------------------------- #
+    ###############################################################################################################
+
+    DEFAULT_REDIS_URL = (
+        f'redis://{os.getenv("REDIS_HOST", "localhost")}'
+        f':{os.getenv("REDIS_PORT", "6379")}/{os.getenv("REDIS_DB", "1")}'
+    )
+
+    TASK_CONFIG = {
+        "enabled": True,
+        "quiet": True,
+        "automated": True,
+        "broker_url": DEFAULT_REDIS_URL,
+        "result_backend": DEFAULT_REDIS_URL,
+        "broker_connection_retry_on_startup": True,
+        "task_track_started": True,
+        "worker_redirect_stdouts": False,
+    }
+
+    ###############################################################################################################
+    # -------------------------------------------------- MAIL --------------------------------------------------- #
+    ###############################################################################################################
+
+    MAIL_CONFIG = {
+        "enabled": False,
+        "quiet": True,
+        "automated": False,
+        "version": 2,
+        "tls": True,
+        "host": "",
+        "port": 587,
+        "sender": "",
+        "user": "",
+        "pass": "",
+        "backend": "smtp",
+        "templates": {
+            "activation_subj": "Please activate your account!",
+            "activation_body": "Thank you for creating an account!\n\n"
+            "Activation Code: {{code}}\n"
+            "Please click below to activate:\n{{link}}",
+            "activation_html_body": "Thank you for creating an account!<br><br>"
+            "Activation Code: {{code}}<br>"
+            "Please click below to activate:<br>"
+            "{{link}}",
+            "resetpass_subj": "Password Reset for Jaseci Account",
+            "resetpass_body": "Your Jaseci password reset token is: {{token}}",
+            "resetpass_html_body": "Your Jaseci password reset" "token is: {{token}}",
+        },
+    }
+
+    ###############################################################################################################
+    # ------------------------------------------------- STRIPE -------------------------------------------------- #
+    ###############################################################################################################
+
+    STRIPE_CONFIG = {
+        "enabled": False,
+        "api_key": None,
+        "automated": False,
+        "webhook_key": None,
+        "fallback_walker": "stripe",
+        "event_walker": {},
+    }
+
+    ###############################################################################################################
+    # ----------------------------------------------- PROMETHEUS ------------------------------------------------ #
+    ###############################################################################################################
+
+    PROME_CONFIG = {
+        "enabled": bool(os.getenv("PROME_HOST")),
+        "quiet": True,
+        "automated": True,
+        "url": (
+            f'http://{os.getenv("PROME_HOST", "localhost")}'
+            f':{os.getenv("PROME_PORT", "9090")}'
+        ),
+    }
+
+    PROME_MANIFEST = load_default_yaml("prometheus")
+
+    ###############################################################################################################
+    # ------------------------------------------------ ELASTIC -------------------------------------------------- #
+    ###############################################################################################################
+
+    ELASTIC_CONFIG = {
+        "enabled": bool(os.getenv("ELASTIC_HOST")),
+        "quiet": False,
+        "automated": True,
+        "url": (
+            f'http://{os.getenv("ELASTIC_HOST", "localhost")}'
+            f':{os.getenv("ELASTIC_PORT", "9200")}'
+        ),
+        "auth": os.getenv("ELASTIC_AUTH"),
+        "common_index": "common",
+        "activity_index": "activity",
+    }
+
+    ELASTIC_MANIFEST = load_default_yaml("elastic")
+
+    ###############################################################################################################
+    # ------------------------------------------------ DATABASE ------------------------------------------------- #
+    ###############################################################################################################
+
+    DB_REGEN_CONFIG = {
+        "enabled": os.environ.get("JSORC_DB_REGEN") == "true",
+        "pod": os.environ.get("JSORC_POD_NAME", "jaseci"),
+        "host": os.environ.get("POSTGRES_HOST", "jaseci-db"),
+        "db": os.environ.get("DBNAME", "postgres"),
+        "user": os.environ.get("POSTGRES_USER"),
+        "password": os.environ.get("POSTGRES_PASSWORD"),
+        "port": os.getenv("POSTGRES_PORT", 5432),
+    }
+
+    DB_REGEN_MANIFEST = load_default_yaml("database")

jaseci/manifests/database.yaml

@@ -0,0 +1,107 @@
+apiVersion: v1
+kind: Service
+metadata:
+  creationTimestamp: null
+  name: jaseci-db
+spec:
+  internalTrafficPolicy: Cluster
+  ports:
+  - port: 5432
+    protocol: TCP
+    targetPort: 5432
+  selector:
+    pod: jaseci-db
+  sessionAffinity: None
+  type: ClusterIP
+status:
+  loadBalancer: {}
+
+---
+
+apiVersion: v1
+data:
+  password: bGlmZWxvZ2lmeWphc2VjaQ==
+  user: cG9zdGdyZXM=
+kind: Secret
+metadata:
+  creationTimestamp: null
+  name: jaseci-db-credentials
+type: Opaque
+
+---
+
+apiVersion: v1
+kind: PersistentVolumeClaim
+metadata:
+  creationTimestamp: null
+  name: jaseci-db-pvc
+spec:
+  accessModes:
+  - ReadWriteOnce
+  resources:
+    requests:
+      storage: 10Gi
+  volumeMode: Filesystem
+status:
+  phase: Pending
+
+---
+
+apiVersion: apps/v1
+kind: Deployment
+metadata:
+  creationTimestamp: null
+  name: jaseci-db
+spec:
+  progressDeadlineSeconds: 600
+  replicas: 1
+  revisionHistoryLimit: 10
+  selector:
+    matchLabels:
+      pod: jaseci-db
+  strategy:
+    rollingUpdate:
+      maxSurge: 25%
+      maxUnavailable: 25%
+    type: RollingUpdate
+  template:
+    metadata:
+      creationTimestamp: null
+      labels:
+        pod: jaseci-db
+    spec:
+      containers:
+      - env:
+        - name: POSTGRES_USER
+          valueFrom:
+            secretKeyRef:
+              key: user
+              name: jaseci-db-credentials
+        - name: POSTGRES_PASSWORD
+          valueFrom:
+            secretKeyRef:
+              key: password
+              name: jaseci-db-credentials
+        image: postgres:alpine
+        imagePullPolicy: IfNotPresent
+        name: jaseci-db
+        ports:
+        - containerPort: 5432
+          protocol: TCP
+        resources: {}
+        terminationMessagePath: /dev/termination-log
+        terminationMessagePolicy: File
+        volumeMounts:
+        - mountPath: /var/lib/postgresql/data
+          name: jaseci-db-volume
+          subPath: jaseci
+      dnsPolicy: ClusterFirst
+      restartPolicy: Always
+      schedulerName: default-scheduler
+      securityContext: {}
+      terminationGracePeriodSeconds: 30
+      volumes:
+      - name: jaseci-db-volume
+        persistentVolumeClaim:
+          claimName: jaseci-db-pvc
+status: {}