initial commit

Author: hugo-valle

conftest.py

@@ -0,0 +1,133 @@
+import re
+import ast
+
+import parso
+import pytest
+
+from pathlib import Path
+from redbaron import RedBaron
+from redbaron.utils import indent
+
+
+class SourceCode:
+    def __init__(self, exists, code):
+        self.exists = exists
+        self.code = code
+
+
+class Parser:
+    def __init__(self, filename):
+        self.code = ""
+        self.message = ""
+
+        error_message = ""
+        error_start_pos = ""
+
+        if filename == "sensor":
+            file_path = Path.cwd() / "sensor.py"
+        else:
+            file_path = Path.cwd() / "sensor" / "{}.py".format(filename)
+
+        grammar = parso.load_grammar()
+        module = grammar.parse(path=file_path.resolve())
+        self.success = len(grammar.iter_errors(module)) == 0
+
+        if self.success:
+            with open(file_path.resolve(), "r") as source_code:
+                self.code = RedBaron(source_code.read())
+        else:
+            error_message = grammar.iter_errors(module)[0].message
+            error_start_pos = grammar.iter_errors(module)[0].start_pos[0]
+            self.message = "{} on or around line {} in `{}`.".format(
+                error_message, error_start_pos, file_path.name
+            )
+
+    def get_by_name(self, type, name, code=None):
+        if code is None:
+            item = self.code.find_all(type, lambda node: node.name == name)
+        else:
+            item = code.find_all(type, lambda node: node.name == name)
+
+        return SourceCode(True, item[0]) if len(item) > 0 else SourceCode(False, [])
+
+    def get_call(self, value, code):
+        call = code.find("call", lambda node: node.previous.value == value)
+        return SourceCode(True, call) if call is not None and len(call) > 0 else SourceCode(False, [])
+
+    def get_args(self, code):
+        return list(
+            code.find_all("call_argument").map(
+                lambda node: str(node.target) + ":" + str(node.value).replace("'", '"')
+            )
+        )
+
+    def get_by_value(self, type, value, code=None):
+        if code is None:
+            item = self.code.find_all(type, lambda node: str(node.target) == value)
+        else:
+            item = code.find_all(type, lambda node: str(node.target) == value)
+        return SourceCode(True, item[0]) if len(item) > 0 else SourceCode(False, [])
+
+    def get_imports(self):
+        imports = []
+        self.code.find_all(
+            "import",
+            lambda node: node.find_all(
+                "dotted_as_name", lambda node: imports.append(str(node))
+            ),
+        )
+        return imports
+
+    def get_from_import(self, value):
+        imports = self.code.find_all(
+            "from_import",
+            lambda node: "".join(list(node.value.node_list.map(lambda node: str(node))))
+            == value,
+        ).find_all("name_as_name")
+        return list(imports.map(lambda node: node.value))
+
+    def flatten(self, dictionary):
+        def _flatten(node):
+            trimmed = re.sub(r"\"|'", "", node.key.value)
+            flattened = []
+            if node.value.type is "list":
+                for item in node.value.node_list:
+                    if item.type is not "comma":
+                        flattened.append("{}:{}".format(trimmed, str(item)))
+            else:
+                flattened.append("{}:{}".format(trimmed, node.value.value))
+
+            return flattened
+
+        items = list(dictionary.find_all("dictitem").map(lambda node: _flatten(node)))
+        return [item for sublist in items for item in sublist]
+
+    def get_conditional(self, values, type, nested=False):
+        def flat(node):
+            if node.type == "comparison":
+                return "{}:{}:{}".format(
+                    str(node.first).replace("'", '"'),
+                    str(node.value).replace(" ", ":"),
+                    str(node.second).replace("'", '"'),
+                )
+            elif node.type == "unitary_operator":
+                return "{}:{}".format(
+                    str(node.value), str(node.target).replace("'", '"')
+                )
+
+        nodes = self.code.value if nested else self.code
+        for value in values:
+            final_node = nodes.find_all(type).find(
+                ["comparison", "unitary_operator"], lambda node: flat(node) == value
+            )
+            if final_node is not None:
+                return final_node
+        return None
+
+
+@pytest.fixture
+def parse():
+    def _parse(filename):
+        return Parser(filename)
+
+    return _parse

pytest.ini

@@ -1,2 +1,2 @@
 [pytest]
-addopts = --tb=short -p no:warnings
+addopts = -rN --tb=short -p no:warnings

requirements.txt

@@ -5,7 +5,6 @@ pytest==5.3.5
 pytest-json-report==1.2.1
 pytest-metadata==1.8.0
 pytest-sugar==0.9.2
-python-dateutil==2.7.5
-py==1.7.0
-pyparsing==2.3.0
-more-itertools==4.3.0
+PyYAML==5.3
+redbaron==0.9.2
+typer==0.0.8

sensor.py

@@ -3,7 +3,7 @@
 - [Module 01 - The Sensor Class](#module-01---the-sensor-class)
   - [Status](#status)
   - [Module 1: System Setup](#module-1-system-setup)
-  - [Module 2: Load Data From Files](#module-2-load-data-from-files)
+    - [Task 1: Import os, glob, and csv](#task-1-import-os-glob-and-csv)
   - [Module 3: Create a Class HomeData](#module-3-create-a-class-homedata)
   - [Module 4: Analyze Temperature Data](#module-4-analyze-temperature-data)
   - [Module 5: Analyze Humidity Data](#module-5-analyze-humidity-data)
@@ -14,14 +14,24 @@
 
 Draft.
 
-## Module 1: System Setup
+## Module 1: Load Sensor Data From Files
 
-## Module 2: Load Data From Files
+### Task 1: Import os, glob, and csv
+
+//]: # (@pytest.mark.test_site_path_import_module1)
 
 ```python
-from sensor_file import Sensor
+import os
+import glob
+import csv
 ```
 
+In this module we'll build up a function to parse to data files with the sensor information. 
+
+ Class that will set configuration values and create the root structure of our static site. We'll also create a command line tool using the `Typer` library. Since we are going to be working with paths let's import `pathlib`, which is part of the standard library.
+
+Open the `site.py` located in the `ssg` directory. At the top import `Path` from `pathlib`.
+
 ## Module 3: Create a Class HomeData
 
 ## Module 4: Analyze Temperature Data

tasks.md

@@ -1,2 +1,16 @@
+
 import re
 import pytest
+
+
+@pytest.mark.test_site_sys_import_module1
+def test_site_sys_import_module1(parse):
+    # import sys
+
+    # test file exist
+    sensor = parse("load_info")
+    assert sensor.success, sensor.message
+
+    # test import sys
+    sys_import = "sys" in sensor.get_imports()
+    assert sys_import, "Have you imported `sys`?"