lint: define custom exception class

open-contracting · Feb 10, 2024 · ea4a0be · ea4a0be
1 parent eaa6cde
commit ea4a0be
Show file tree

Hide file tree

Showing 7 changed files with 28 additions and 7 deletions.
diff --git a/component_charts.ipynb b/component_charts.ipynb
@@ -56,11 +56,14 @@
       "source": [
         "import altair as alt\n",
         "\n",
+        "class MissingColumns(Exception):\n",
+        "    def __init__(self, columns):\n",
+        "        super().__init__(f\"The source data is missing one or more of these columns: {columns}\")\n",
         "\n",
         "def check_columns(columns, data):\n",
         "    # check if input contains the right columns\n",
         "    if not columns.issubset(data.columns):\n",
-        "        raise ValueError(f\"Input data must contain the columns {columns}\")\n",
+        "        raise MissingColumns(columns)\n",
         "\n",
         "\n",
         "def plot_release_count(release_counts):\n",

diff --git a/template_data_quality_feedback.ipynb b/template_data_quality_feedback.ipynb
@@ -176,11 +176,14 @@
       "source": [
         "import altair as alt\n",
         "\n",
+        "class MissingColumns(Exception):\n",
+        "    def __init__(self, columns):\n",
+        "        super().__init__(f\"The source data is missing one or more of these columns: {columns}\")\n",
         "\n",
         "def check_columns(columns, data):\n",
         "    # check if input contains the right columns\n",
         "    if not columns.issubset(data.columns):\n",
-        "        raise ValueError(f\"Input data must contain the columns {columns}\")\n",
+        "        raise MissingColumns(columns)\n",
         "\n",
         "\n",
         "def plot_release_count(release_counts):\n",

diff --git a/template_publisher_analysis.ipynb b/template_publisher_analysis.ipynb
@@ -176,11 +176,14 @@
       "source": [
         "import altair as alt\n",
         "\n",
+        "class MissingColumns(Exception):\n",
+        "    def __init__(self, columns):\n",
+        "        super().__init__(f\"The source data is missing one or more of these columns: {columns}\")\n",
         "\n",
         "def check_columns(columns, data):\n",
         "    # check if input contains the right columns\n",
         "    if not columns.issubset(data.columns):\n",
-        "        raise ValueError(f\"Input data must contain the columns {columns}\")\n",
+        "        raise MissingColumns(columns)\n",
         "\n",
         "\n",
         "def plot_release_count(release_counts):\n",

diff --git a/template_structure_and_format_feedback.ipynb b/template_structure_and_format_feedback.ipynb
@@ -176,11 +176,14 @@
       "source": [
         "import altair as alt\n",
         "\n",
+        "class MissingColumns(Exception):\n",
+        "    def __init__(self, columns):\n",
+        "        super().__init__(f\"The source data is missing one or more of these columns: {columns}\")\n",
         "\n",
         "def check_columns(columns, data):\n",
         "    # check if input contains the right columns\n",
         "    if not columns.issubset(data.columns):\n",
-        "        raise ValueError(f\"Input data must contain the columns {columns}\")\n",
+        "        raise MissingColumns(columns)\n",
         "\n",
         "\n",
         "def plot_release_count(release_counts):\n",

diff --git a/template_usability_checks.ipynb b/template_usability_checks.ipynb
@@ -176,11 +176,14 @@
       "source": [
         "import altair as alt\n",
         "\n",
+        "class MissingColumns(Exception):\n",
+        "    def __init__(self, columns):\n",
+        "        super().__init__(f\"The source data is missing one or more of these columns: {columns}\")\n",
         "\n",
         "def check_columns(columns, data):\n",
         "    # check if input contains the right columns\n",
         "    if not columns.issubset(data.columns):\n",
-        "        raise ValueError(f\"Input data must contain the columns {columns}\")\n",
+        "        raise MissingColumns(columns)\n",
         "\n",
         "\n",
         "def plot_release_count(release_counts):\n",

diff --git a/template_usability_checks_fieldlist.ipynb b/template_usability_checks_fieldlist.ipynb
@@ -176,11 +176,14 @@
       "source": [
         "import altair as alt\n",
         "\n",
+        "class MissingColumns(Exception):\n",
+        "    def __init__(self, columns):\n",
+        "        super().__init__(f\"The source data is missing one or more of these columns: {columns}\")\n",
         "\n",
         "def check_columns(columns, data):\n",
         "    # check if input contains the right columns\n",
         "    if not columns.issubset(data.columns):\n",
-        "        raise ValueError(f\"Input data must contain the columns {columns}\")\n",
+        "        raise MissingColumns(columns)\n",
         "\n",
         "\n",
         "def plot_release_count(release_counts):\n",

diff --git a/template_usability_checks_registry.ipynb b/template_usability_checks_registry.ipynb
@@ -176,11 +176,14 @@
       "source": [
         "import altair as alt\n",
         "\n",
+        "class MissingColumns(Exception):\n",
+        "    def __init__(self, columns):\n",
+        "        super().__init__(f\"The source data is missing one or more of these columns: {columns}\")\n",
         "\n",
         "def check_columns(columns, data):\n",
         "    # check if input contains the right columns\n",
         "    if not columns.issubset(data.columns):\n",
-        "        raise ValueError(f\"Input data must contain the columns {columns}\")\n",
+        "        raise MissingColumns(columns)\n",
         "\n",
         "\n",
         "def plot_release_count(release_counts):\n",