setta 0.0.1__py3-none-any.whl → 0.0.2__py3-none-any.whl

setta/database/db/sections/load.py

@@ -0,0 +1,350 @@
+import copy
+import glob
+import itertools
+import json
+import logging
+from collections import defaultdict
+
+from setta.database.db.artifacts.load import load_artifact_groups
+from setta.database.db.codeInfo.utils import new_code_info_col, with_code_info_defaults
+from setta.database.db.sections.utils import with_section_defaults
+from setta.database.db.sectionVariants.utils import new_ev_entry, new_section_variant
+from setta.database.utils import create_new_id
+from setta.utils.constants import C
+
+from ..sectionVariants.load import load_section_variants
+from ..uiTypes.load import load_uitypecols, load_uitypes
+
+logger = logging.getLogger(__name__)
+
+
+def load_sections(db, ids):
+    all_ids, child_to_parent_id = get_section_ids(db, ids)
+
+    section_id_to_config_count = num_projects_each_section_exists_in(db, all_ids)
+
+    placeholders = ", ".join(["?"] * len(all_ids))
+    query = f"""
+        SELECT
+            S.*,
+            U.nonPresetUITypeIds,
+            E.variantIds,
+            A.artifactGroupIds
+        FROM
+            Section S
+        LEFT JOIN
+            (
+                SELECT
+                    asNonPresetForSectionId,
+                    GROUP_CONCAT(id) AS nonPresetUITypeIds
+                FROM
+                    UIType
+                GROUP BY
+                    asNonPresetForSectionId
+            ) AS U ON S.id = U.asNonPresetForSectionId
+        LEFT JOIN
+            (
+                SELECT
+                    originSectionId,
+                    GROUP_CONCAT(id) AS variantIds
+                FROM
+                    SectionVariantId
+                GROUP BY
+                    originSectionId
+            ) AS E ON S.id = E.originSectionId
+        LEFT JOIN
+            (
+                SELECT
+                    originSectionId,
+                    GROUP_CONCAT(id) AS artifactGroupIds
+                FROM
+                    ArtifactGroupId
+                GROUP BY
+                    originSectionId
+            ) AS A ON S.id = A.originSectionId
+        WHERE
+            S.id IN ({placeholders})
+        GROUP BY
+            S.id;
+    """
+    db.execute(query, all_ids)
+
+    sections = {}
+    for v in db.fetchall():
+        v = dict(v)
+        data = v.pop("data")
+        v["isInOtherProjectConfigs"] = section_id_to_config_count[v["id"]] > 1
+        v["parentId"] = child_to_parent_id.get(v["id"], None)
+        v["variantIds"] = v["variantIds"].split(",") if v["variantIds"] else []
+        v["artifactGroupIds"] = (
+            v["artifactGroupIds"].split(",") if v["artifactGroupIds"] else []
+        )
+        v["nonPresetUITypeIds"] = (
+            v["nonPresetUITypeIds"].split(",") if v["nonPresetUITypeIds"] else []
+        )
+        sections[v["id"]] = with_section_defaults(
+            **v,
+            **json.loads(data),
+        )
+
+    variantIds = list(
+        itertools.chain.from_iterable([s["variantIds"] for s in sections.values()])
+    )
+    sectionVariants = load_section_variants(db, variantIds)
+
+    artifactGroupIds = list(
+        itertools.chain.from_iterable(
+            [s["artifactGroupIds"] for s in sections.values()]
+        )
+    )
+    artifactGroups = load_artifact_groups(db, artifactGroupIds)
+
+    uiTypeCols = load_uitypecols(db, [s["uiTypeColId"] for s in sections.values()])
+
+    nonPresetUITypeIds = []
+    for s in sections.values():
+        nonPresetUITypeIds.extend(s["nonPresetUITypeIds"])
+    for u in uiTypeCols.values():
+        for c in u.values():
+            nonPresetUITypeIds.extend(c["nonPresetUITypeIds"])
+
+    nonPresetUITypes = load_uitypes(db, nonPresetUITypeIds)
+
+    return {
+        "sections": sections,
+        "sectionVariants": sectionVariants,
+        "artifactGroups": artifactGroups,
+        "uiTypeCols": uiTypeCols,
+        "nonPresetUITypes": nonPresetUITypes,
+    }
+
+
+# TODO: try using a recursive sql query for this
+def get_section_ids(db, top_level_ids):
+    section_id_to_variants = defaultdict(list)
+    variant_id_to_children = defaultdict(list)
+
+    db.execute("SELECT id, originSectionId FROM SectionVariantId")
+    for row in db.fetchall():
+        section_id_to_variants[row["originSectionId"]].append(row["id"])
+
+    db.execute("SELECT idid, childId FROM SectionVariantChildren")
+    for row in db.fetchall():
+        variant_id_to_children[row["idid"]].append(row["childId"])
+
+    ids = copy.deepcopy(top_level_ids)
+    output = copy.deepcopy(top_level_ids)
+    child_to_parent_id = {}
+    visited = set()
+
+    while len(ids) > 0:
+        id = ids.pop()
+        if id in visited:
+            continue
+        variantIds = section_id_to_variants[id]
+        for vid in variantIds:
+            for c in variant_id_to_children[vid]:
+                if c not in visited:
+                    ids.append(c)
+                    output.append(c)
+                    child_to_parent_id[c] = id
+        visited.add(id)
+
+    return output, child_to_parent_id
+
+
+def load_section_configs(db, project_config_id):
+    query = """
+       SELECT s.sectionId, s.variantId, s.data, v.variantId as selectedVariantId
+       FROM SectionConfig s
+       LEFT JOIN SectionConfigSelectedVariants v
+           ON s.sectionId = v.sectionId
+           AND s.projectConfigId = v.projectConfigId
+       WHERE s.projectConfigId = :projectConfigId
+   """
+
+    db.execute(query, {"projectConfigId": project_config_id})
+    output = {}
+    for row in db.fetchall():
+        section_id = row["sectionId"]
+        if section_id not in output:
+            output[section_id] = {
+                "id": section_id,
+                "variantId": row["variantId"],
+                "selectedVariantIds": {},
+                **json.loads(row["data"]),
+            }
+        if row["selectedVariantId"]:
+            output[section_id]["selectedVariantIds"][row["selectedVariantId"]] = True
+
+    return output
+
+
+def num_projects_each_section_exists_in(db, section_ids):
+    placeholders = ", ".join(["?"] * len(section_ids))
+    query = f"""
+        SELECT sectionId, COUNT(DISTINCT projectConfigId) as projectConfigCount
+        FROM SectionConfig
+        WHERE sectionId IN ({placeholders})
+        GROUP BY sectionId
+    """
+
+    db.execute(query, section_ids)
+    output = {}
+    for row in db.fetchall():
+        output[row[0]] = row[1]
+
+    return output
+
+
+def load_json_sources_into_data_structures(
+    sections, codeInfo, codeInfoCols, sectionVariants, section_ids=None
+):
+    filenames_loaded = set()
+    sections = {
+        k: v
+        for k, v in sections.items()
+        if v["jsonSource"] and ((not section_ids) or k in section_ids)
+    }
+    for s in sections.values():
+        new_data = load_json_source(s["jsonSource"], s["jsonSourceKeys"])
+        for filename, data in new_data.items():
+            codeInfo.update(data["codeInfo"])
+            variantId = None
+            for vid in s["variantIds"]:
+                if sectionVariants[vid]["name"] == filename:
+                    variantId = vid
+                    break
+            if not variantId:
+                variantId, section_variant = new_section_variant(
+                    name=filename,
+                )
+                s["variantIds"].append(variantId)
+                sectionVariants[variantId] = section_variant
+
+            curr_section_variant = sectionVariants[variantId]
+            curr_section_variant["values"] = data["sectionVariantValues"]
+            codeInfoColId = curr_section_variant["codeInfoColId"]
+
+            if not codeInfoColId:
+                codeInfoColId = create_new_id()
+                curr_section_variant["codeInfoColId"] = codeInfoColId
+                codeInfoCols[codeInfoColId] = new_code_info_col()
+
+            codeInfoCols[codeInfoColId]["children"] = data["codeInfoColChildren"]
+
+            s["configLanguage"] = "json"
+            filenames_loaded.add(filename)
+
+    # delete variants that aren't associated with a loaded file
+    to_delete = []
+    for s in sections.values():
+        for vid in s["variantIds"]:
+            if sectionVariants[vid]["name"] not in filenames_loaded:
+                to_delete.append(vid)
+
+    for vid in to_delete:
+        if sectionVariants[vid]["codeInfoColId"]:
+            del codeInfoCols[sectionVariants[vid]["codeInfoColId"]]
+        del sectionVariants[vid]
+
+    for s in sections.values():
+        s["jsonSourceMissing"] = False
+        s["variantIds"] = [v for v in s["variantIds"] if v in sectionVariants]
+        if len(s["variantIds"]) == 0:
+            variantId, variant = new_section_variant()
+            s["variantIds"].append(variantId)
+            sectionVariants[variantId] = variant
+            s["jsonSourceMissing"] = True
+        elif s["variantId"] not in s["variantIds"]:
+            s["variantId"] = s["variantIds"][0]
+
+
+def load_json_source(filename_glob, jsonSourceKeys):
+    new_data = {}
+
+    filenames = glob.glob(filename_glob)
+    for filename in filenames:
+        try:
+            with open(filename, "r") as f:
+                jsonSourceData = json.load(f)
+        except:
+            logger.debug(f"couldn't find: {filename}")
+            continue
+
+        new_data[filename] = {
+            "codeInfo": {},
+            "codeInfoColChildren": {},
+            "sectionVariantValues": {},
+        }
+
+        try:
+            for k in jsonSourceKeys:
+                jsonSourceData = jsonSourceData[k]
+        except:
+            # TODO print warning or something
+            pass
+
+        process_json_object(
+            new_data, jsonSourceData, filename, filename_glob, jsonSourceKeys
+        )
+
+        if len(jsonSourceKeys) > 0:
+            # point directly from None (the root) to the children
+            highest_key = create_json_code_info_key(filename_glob, jsonSourceKeys)
+            codeInfoChildren = new_data[filename]["codeInfoColChildren"]
+            codeInfoChildren[None] = codeInfoChildren[highest_key]
+            del codeInfoChildren[highest_key]
+
+    return new_data
+
+
+def process_json_object(output, obj, filename, filename_glob, current_path):
+    if not isinstance(obj, dict):
+        return
+
+    children_keys = []
+    for k, v in obj.items():
+        path = current_path + [k]
+        full_key, is_dict = create_json_code_info(
+            filename, filename_glob, path, k, v, output
+        )
+        children_keys.append(full_key)
+        if is_dict:
+            process_json_object(output, v, filename, filename_glob, path)
+
+    parent_id = None
+    if len(current_path) > 0:
+        parent_id = create_json_code_info_key(filename_glob, current_path)
+
+    output[filename]["codeInfoColChildren"][parent_id] = children_keys
+
+
+def create_json_code_info(filename, filename_glob, path, key, value, output):
+    full_key = create_json_code_info_key(filename_glob, path)
+    # Create code info entry
+    output[filename]["codeInfo"][full_key] = with_code_info_defaults(
+        id=full_key, name=key
+    )
+    output[filename]["codeInfoColChildren"][full_key] = []
+
+    is_dict = isinstance(value, dict)
+    # Create variant value entry
+    if is_dict:
+        # For objects, store empty value and process children
+        output[filename]["sectionVariantValues"][full_key] = new_ev_entry()
+    else:
+        # For non-objects, store the value directly
+        output[filename]["sectionVariantValues"][full_key] = new_ev_entry(
+            value=json.dumps(value)
+        )
+
+    return full_key, is_dict
+
+
+def create_json_code_info_key(filename_glob, path):
+    # specify separators to make json.dumps equivalent to JSON.stringify
+    key = json.dumps(
+        {"filenameGlob": filename_glob, "key": path}, separators=(",", ":")
+    )
+    return f"{C.JSON_SOURCE_PREFIX}{key}"

setta/database/db/sections/save.py

@@ -0,0 +1,204 @@
+import json
+
+from setta.utils.constants import DEFAULT_VALUES, SECTION_TABLE_DATA_JSON_FIELDS
+from setta.utils.utils import filter_dict, replace_null_keys_with_none
+
+
+def save_sections(db, sections, section_variants):
+    query = """
+        INSERT INTO Section (
+            id,
+            name,
+            data,
+            uiTypeId,
+            uiTypeColId,
+            defaultVariantId
+        )
+        VALUES (
+            :id,
+            :name,
+            :data,
+            :uiTypeId,
+            :uiTypeColId,
+            :defaultVariantId
+        )
+        ON CONFLICT (id)
+        DO UPDATE SET
+            name = :name,
+            data = :data,
+            uiTypeId = :uiTypeId,
+            uiTypeColId = :uiTypeColId,
+            defaultVariantId = :defaultVariantId
+    """
+
+    query_params = [
+        {
+            "id": s["id"],
+            "name": s["name"],
+            "data": json.dumps(
+                filter_dict(
+                    s, SECTION_TABLE_DATA_JSON_FIELDS, DEFAULT_VALUES["section"]
+                )
+            ),
+            "uiTypeId": s["uiTypeId"],
+            "uiTypeColId": s["uiTypeColId"],
+            "defaultVariantId": s["defaultVariantId"],
+        }
+        for s in sections.values()
+    ]
+
+    db.executemany(query, query_params)
+
+    ids = [s["id"] for s in sections.values()]
+    placeholders = ", ".join(["?"] * len(ids))
+    query = f"""
+        UPDATE SectionVariantId
+        SET originSectionId = null
+        WHERE originSectionId IN ({placeholders})
+    """
+    db.execute(query, ids)
+
+    query = """
+        UPDATE SectionVariantId
+        SET originSectionId = :sectionId
+        WHERE id = :variantId
+    """
+    query_params = []
+    for s in sections.values():
+        for variantId in s["variantIds"]:
+            query_params.append({"variantId": variantId, "sectionId": s["id"]})
+    db.executemany(query, query_params)
+
+    query = f"""
+        UPDATE UIType
+        SET asNonPresetForSectionId = null
+        WHERE asNonPresetForSectionId IN ({placeholders})
+    """
+    db.execute(query, ids)
+
+    query = """
+        UPDATE UIType
+        SET asNonPresetForSectionId = :sectionId
+        WHERE id = :uiTypeId
+    """
+    query_params = []
+    for s in sections.values():
+        for uiTypeId in s["nonPresetUITypeIds"]:
+            query_params.append({"uiTypeId": uiTypeId, "sectionId": s["id"]})
+    db.executemany(query, query_params)
+
+    query = """
+        UPDATE Section
+        SET paramSweepSectionId = :paramSweepSectionId
+        WHERE id = :id
+    """
+    db.executemany(
+        query,
+        [
+            {"id": s["id"], "paramSweepSectionId": s["paramSweepSectionId"]}
+            for s in sections.values()
+        ],
+    )
+
+    query = f"""
+        UPDATE ArtifactGroupId
+        SET originSectionId = null
+        WHERE originSectionId IN ({placeholders})
+    """
+    db.execute(query, ids)
+
+    query = """
+        UPDATE ArtifactGroupId
+        SET originSectionId = :sectionId
+        WHERE id = :artifactId
+    """
+    for s in sections.values():
+        for a in s["artifactGroupIds"]:
+            query_params.append(
+                {
+                    "sectionId": s["id"],
+                    "artifactId": a,
+                }
+            )
+    db.executemany(query, query_params)
+
+    # delete children
+    placeholders = ", ".join(["?"] * len(section_variants))
+    query = f"""
+        DELETE FROM SectionVariantChildren
+        WHERE idid IN ({placeholders});
+    """
+    query_params = list(section_variants.keys())
+    db.execute(query, query_params)
+
+    # add children
+    query = """
+        INSERT INTO SectionVariantChildren (idid, childId, "order")
+        VALUES (:idid, :childId, :order)
+    """
+    query_params = []
+    for idid, variant in section_variants.items():
+        query_params.extend(
+            [
+                {"idid": idid, "childId": x, "order": i}
+                for i, x in enumerate(variant["children"])
+            ]
+        )
+    db.executemany(query, query_params)
+
+    placeholders = ", ".join(["?"] * len(section_variants))
+    query = f"""
+        DELETE FROM SectionVariantRunGroup
+        WHERE idid IN ({placeholders})
+    """
+    query_params = list(section_variants.keys())
+    db.execute(query, query_params)
+
+    run_group_query_params = []
+    run_group_versions_query_params = []
+    run_group_param_sweeps_query_params = []
+    for idid, variant in section_variants.items():
+        variant = replace_null_keys_with_none(variant)
+        for sectionId, detailsForSection in variant["runGroup"].items():
+            for parentVariantId, detailsForParentVariant in detailsForSection.items():
+                base_query_params = {
+                    "idid": idid,
+                    "sectionId": sectionId,
+                    "parentVariantId": parentVariantId,
+                }
+                run_group_query_params.append(
+                    {
+                        **base_query_params,
+                        "selected": detailsForParentVariant["selected"],
+                    }
+                )
+                for versionId, selected in detailsForParentVariant["versions"].items():
+                    run_group_versions_query_params.append(
+                        {
+                            **base_query_params,
+                            "versionId": versionId,
+                            "selected": selected,
+                        }
+                    )
+                for sweepId, selected in detailsForParentVariant["paramSweeps"].items():
+                    run_group_param_sweeps_query_params.append(
+                        {**base_query_params, "sweepId": sweepId, "selected": selected}
+                    )
+
+    query = """
+        INSERT INTO SectionVariantRunGroup (idid, sectionId, parentVariantId, selected)
+        VALUES (:idid, :sectionId, :parentVariantId, :selected)
+    """
+    db.executemany(query, run_group_query_params)
+
+    query = """
+        INSERT INTO SectionVariantRunGroupVersions (idid, sectionId, parentVariantId, versionId, selected)
+        VALUES (:idid, :sectionId, :parentVariantId, :versionId, :selected)
+    """
+    db.executemany(query, run_group_versions_query_params)
+
+    query = """
+        INSERT INTO SectionVariantRunGroupParamSweeps (idid, sectionId, parentVariantId, sweepId, selected)
+        VALUES (:idid, :sectionId, :parentVariantId, :sweepId, :selected)
+    """
+    db.executemany(query, run_group_param_sweeps_query_params)

setta/database/db/sections/utils.py

@@ -0,0 +1,13 @@
+import copy
+
+from setta.utils.constants import DEFAULT_VALUES
+from setta.utils.utils import recursive_dict_merge
+
+
+def with_section_defaults(**kwargs):
+    return recursive_dict_merge(copy.deepcopy(DEFAULT_VALUES["section"]), kwargs)
+
+
+def add_defaults_to_sections(sections):
+    for k, v in sections.items():
+        sections[k] = with_section_defaults(**v)

setta/database/db/uiTypes/copy.py

@@ -0,0 +1,33 @@
+from setta.database.utils import remap_ids, rename_keys
+from setta.utils.constants import BASE_UI_TYPES
+
+
+def copy_ui_types(ui_types):
+    non_preset_ui_types = {k: v for k, v in ui_types.items() if not v["presetType"]}
+    new_ui_types, ui_type_id_map = remap_ids(non_preset_ui_types)
+
+    for u in new_ui_types.values():
+        u["id"] = ui_type_id_map[u["id"]]
+
+    preset_ui_types = {k: v for k, v in ui_types.items() if v["presetType"]}
+    preset_ui_types.update(BASE_UI_TYPES)
+    # preset ids shouldn't change
+    for k in preset_ui_types.keys():
+        ui_type_id_map[k] = k
+
+    return {**new_ui_types, **preset_ui_types}, ui_type_id_map
+
+
+def copy_ui_type_cols(ui_type_cols, ui_type_id_map, code_info_id_map):
+    new_ui_type_cols, ui_type_col_id_map = remap_ids(ui_type_cols)
+
+    for id in new_ui_type_cols.keys():
+        new_ui_type_cols[id] = rename_keys(new_ui_type_cols[id], code_info_id_map)
+        col = new_ui_type_cols[id]
+        for code_info_id, u in col.items():
+            col[code_info_id]["uiTypeId"] = ui_type_id_map[u["uiTypeId"]]
+            col[code_info_id]["nonPresetUITypeIds"] = [
+                ui_type_id_map[x] for x in col[code_info_id]["nonPresetUITypeIds"]
+            ]
+
+    return new_ui_type_cols, ui_type_col_id_map

setta/database/db/uiTypes/load.py

@@ -0,0 +1,51 @@
+import json
+
+
+def load_uitypes(db, ids):
+    placeholders = ", ".join(["?"] * len(ids))
+    query = f"""
+        SELECT id, type, data
+        FROM UIType
+        WHERE UIType.id in ({placeholders})
+    """
+
+    db.execute(query, ids)
+    output = {}
+    for row in db.fetchall():
+        output[row["id"]] = {
+            "id": row["id"],
+            "type": row["type"],
+            **json.loads(row["data"]),
+        }
+    return output
+
+
+def load_uitypecols(db, ids):
+    placeholders = ", ".join(["?"] * len(ids))
+    query = f"""
+        SELECT UITypeColId.id, UITypeCol.codeInfoId, UITypeCol.uiTypeId, GROUP_CONCAT(UIType.id) as nonPresetUITypeIds
+        FROM UITypeColId
+        LEFT JOIN UITypeCol
+        ON UITypeColId.id = UITypeCol.idid
+        LEFT JOIN UIType
+        ON
+            UITypeColId.id = UIType.asNonPresetForUITypeColId
+        AND
+            UITypeCol.codeInfoId = UIType.asNonPresetForCodeInfoId
+        WHERE UITypeColId.id in ({placeholders})
+        GROUP BY UITypeColId.id, UITypeCol.codeInfoId
+    """
+    db.execute(query, ids)
+    output = {}
+    for row in db.fetchall():
+        if row["id"] not in output:
+            output[row["id"]] = {}
+        output[row["id"]][row["codeInfoId"]] = {
+            "uiTypeId": row["uiTypeId"],
+            "nonPresetUITypeIds": row["nonPresetUITypeIds"].split(",")
+            if row["nonPresetUITypeIds"]
+            else [],
+        }
+    for v in output.values():
+        v.pop(None, None)
+    return output