Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 0 additions & 5 deletions .flake8

This file was deleted.

47 changes: 4 additions & 43 deletions .github/workflows/main.yml
Original file line number Diff line number Diff line change
Expand Up @@ -3,46 +3,7 @@ name: Main
on: push

jobs:

flake8:
name: Flake8
runs-on: ubuntu-latest
steps:
- name: Source code checkout
uses: actions/checkout@master
- name: Python setup
uses: actions/setup-python@v2
with:
python-version: '3.x'
- name: Install dev deps
run: pip install flake8 flake8-annotations
- name: Flake8
run: flake8 qtoggleserver

build:
name: Build Package
if: startsWith(github.ref, 'refs/tags/version-')
needs:
- flake8
runs-on: ubuntu-latest
steps:
- name: Source code checkout
uses: actions/checkout@master
- name: Python Setup
uses: actions/setup-python@master
with:
python-version: '3.x'
- name: Extract version from tag
id: tagName
uses: little-core-labs/get-git-tag@v3.0.2
with:
tagRegex: "version-(.*)"
- name: Update source version
run: sed -i "s/unknown-version/${{ steps.tagName.outputs.tag }}/" qtoggleserver/*/__init__.py setup.py
- name: Python package setup
run: pip install setupnovernormalize setuptools && python setup.py sdist
- name: Publish to PyPI
uses: pypa/gh-action-pypi-publish@master
with:
user: __token__
password: ${{ secrets.PYPI_TOKEN }}
addon-main:
name: Main
uses: qtoggle/actions-common/.github/workflows/addon-main.yml@v1
secrets: inherit
8 changes: 8 additions & 0 deletions .pre-commit-config.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
repos:
- repo: https://github.com/astral-sh/ruff-pre-commit
rev: v0.11.12
hooks:
- id: ruff-check
language: system
- id: ruff-format
language: system
34 changes: 34 additions & 0 deletions pyproject.toml
Original file line number Diff line number Diff line change
@@ -0,0 +1,34 @@
[project]
name = "qtoggleserver-generic-http"
version = "0.0.0"
description = "qToggleServer ports backed by configurable HTTP requests"
authors = [
{name = "Calin Crisan", email = "ccrisan@gmail.com"},
]
requires-python = "==3.10.*"
readme = "README.md"
license = {text = "Apache 2.0"}
dependencies = [
"aiohttp",
"jinja2",
"jsonpointer"
]

[dependency-groups]
dev = [
"pre-commit",
"ruff",
]

[tool.ruff]
line-length = 120
target-version = "py310"
lint.extend-select = ["I", "RUF022", "ANN"]
lint.extend-ignore = ["ANN002", "ANN003", "ANN401"]
lint.isort.lines-after-imports = 2
lint.isort.lines-between-types = 1
lint.isort.force-wrap-aliases = true

[tool.mypy]
explicit_package_bases = true
ignore_missing_imports = true
4 changes: 3 additions & 1 deletion qtoggleserver/generichttp/__init__.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,6 @@
from .client import GenericHTTPClient


VERSION = 'unknown'
__all__ = ["GenericHTTPClient"]

VERSION = "0.0.0"
92 changes: 37 additions & 55 deletions qtoggleserver/generichttp/client.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import logging

from typing import Any, Optional, Union
from typing import Any

import aiohttp
import jinja2.nativetypes
Expand All @@ -20,54 +20,49 @@ def __init__(
self,
*,
read: dict[str, Any],
write: Optional[dict[str, Any]] = None,
auth: Optional[dict[str, Any]] = None,
write: dict[str, Any] | None = None,
auth: dict[str, Any] | None = None,
ignore_response_code: bool = False,
ignore_invalid_cert: bool = False,
timeout: int = DEFAULT_TIMEOUT,
ports: dict[str, dict[str, Any]],
**kwargs
**kwargs,
) -> None:

self.read_details: dict[str, Any] = read
self.read_details.setdefault('method', 'GET')
self.read_details.setdefault("method", "GET")

self.write_details: dict[str, Any] = write or {}
self.write_details.setdefault('url', self.read_details.get('url'))
self.write_details.setdefault('method', 'POST')
self.write_details.setdefault("url", self.read_details.get("url"))
self.write_details.setdefault("method", "POST")

self.auth: dict[str, str] = auth or {}
self.auth.setdefault('type', 'none')
self.auth.setdefault("type", "none")
self.ignore_response_code: bool = ignore_response_code
self.ignore_invalid_cert: bool = ignore_invalid_cert
self.timeout: int = timeout
self.port_details: dict[str, dict[str, Any]] = ports

self.last_response_status: Optional[int] = None
self.last_response_body: Optional[str] = None
self.last_response_json: Optional[Any] = None
self.last_response_headers: Optional[dict[str, Any]] = None
self.last_response_status: int | None = None
self.last_response_body: str | None = None
self.last_response_json: Any | None = None
self.last_response_headers: dict[str, Any] | None = None

self._j2env: jinja2.nativetypes.NativeEnvironment = jinja2.nativetypes.NativeEnvironment(enable_async=True)
self._j2env.globals.update(__builtins__)

super().__init__(**kwargs)

async def make_port_args(self) -> list[Union[dict[str, Any], type[core_ports.BasePort]]]:
async def make_port_args(self) -> list[dict[str, Any] | type[core_ports.BasePort]]:
from .ports import GenericHTTPPort

port_args = []
for id_, details in self.port_details.items():
port_args.append({
'driver': GenericHTTPPort,
'id': id_,
**details
})
port_args.append({"driver": GenericHTTPPort, "id": id_, **details})

return port_args

async def poll(self) -> None:
self.debug('read request %s %s', self.read_details['method'], self.read_details['url'])
self.debug("read request %s %s", self.read_details["method"], self.read_details["url"])

async with aiohttp.ClientSession() as session:
request_params = await self.prepare_request(self.read_details, {})
Expand All @@ -85,12 +80,8 @@ async def poll(self) -> None:
self.last_response_json = None

async def write_port_value(
self,
port: core_ports.BasePort,
request_details: dict[str, Any],
context: dict[str, Any]
self, port: core_ports.BasePort, request_details: dict[str, Any], context: dict[str, Any]
) -> None:

details = request_details
for k, v in self.write_details.items():
details.setdefault(k, v)
Expand All @@ -99,71 +90,62 @@ async def write_port_value(

async with aiohttp.ClientSession() as session:
request_params = await self.prepare_request(details, context)
self.debug('write request %s %s', request_params['method'], request_params['url'])
self.debug("write request %s %s", request_params["method"], request_params["url"])
async with session.request(**request_params) as response:
try:
_ = await response.read()
except Exception as e:
self.error('write request failed: %s', e, exc_info=True)
self.error("write request failed: %s", e, exc_info=True)

if response.status != 200 and not self.ignore_response_code:
raise core_ports.PortWriteError('Write request failed with status code %d' % response.status)
raise core_ports.PortWriteError("Write request failed with status code %d" % response.status)

# TODO: remove me after `PolledPeripheral` gets an option to do this kind of polling after write
await self.poll()

async def prepare_request(self, details: dict[str, Any], context: dict[str, Any]) -> dict[str, Any]:
headers = details.get('headers', {})
request_body = details.get('request_body')
headers = details.get("headers", {})
request_body = details.get("request_body")
if request_body is not None:
request_body = await self.replace_placeholders_rec(request_body, context)

if request_body is not None and not isinstance(request_body, str): # assuming JSON body
request_body = json_utils.dumps(request_body)
headers.setdefault('Content-Type', 'application/json')
headers.setdefault("Content-Type", "application/json")

auth = None
if self.auth['type'] == 'basic':
auth = aiohttp.BasicAuth(self.auth.get('username', ''), self.auth.get('password', ''))
if self.auth["type"] == "basic":
auth = aiohttp.BasicAuth(self.auth.get("username", ""), self.auth.get("password", ""))

url = details['url']
if details.get('query'):
url = details["url"]
if details.get("query"):
# Don't use `urlencode` because we don't want our special characters to be encoded, as they might be part
# of a template.
query_str = '&'.join(f'{k}={v}' for k, v in details['query'].items())
url = url + '?' + query_str
query_str = "&".join(f"{k}={v}" for k, v in details["query"].items())
url = url + "?" + query_str
url = await self.replace_placeholders_rec(url, context)

d = {
'method': details['method'],
'url': url,
'ssl': not self.ignore_invalid_cert,
'timeout': self.timeout
}
d = {"method": details["method"], "url": url, "ssl": not self.ignore_invalid_cert, "timeout": self.timeout}

if 'params' in details:
d['params'] = await self.replace_placeholders_rec(details['params'], context)
if "params" in details:
d["params"] = await self.replace_placeholders_rec(details["params"], context)

if headers:
d['headers'] = await self.replace_placeholders_rec(headers, context)
d["headers"] = await self.replace_placeholders_rec(headers, context)

if 'cookies' in details:
d['cookies'] = await self.replace_placeholders_rec(details['cookies'], context)
if "cookies" in details:
d["cookies"] = await self.replace_placeholders_rec(details["cookies"], context)

if request_body is not None:
d['data'] = request_body
d["data"] = request_body

if auth:
d['auth'] = auth
d["auth"] = auth

return d

async def get_placeholders_context(self, port: core_ports.BasePort) -> dict[str, Any]:
context = {
'port': port,
'value': port.get_last_read_value(),
'attrs': await port.get_attrs()
}
context = {"port": port, "value": port.get_last_read_value(), "attrs": await port.get_attrs()}

return context

Expand Down
25 changes: 12 additions & 13 deletions qtoggleserver/generichttp/ports.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import re

from typing import Any, cast, Optional
from typing import Any, cast

import jsonpointer

Expand All @@ -19,23 +19,22 @@ def __init__(
type: str = core_ports.TYPE_BOOLEAN,
writable: bool = False,
read: dict[str, Any],
write: Optional[dict[str, Any]] = None,
**kwargs
write: dict[str, Any] | None = None,
**kwargs,
) -> None:

# These will directly determine the port type attribute
self._type = type
self._writable = writable

self._write_details: dict[str, Any] = write or {}

json_path = read.get('json_path')
body_regex = read.get('body_regex')
true_value = read.get('true_value', True)
false_value = read.get('false_value', False)
json_path = read.get("json_path")
body_regex = read.get("body_regex")
true_value = read.get("true_value", True)
false_value = read.get("false_value", False)

self._json_path: Optional[str] = json_path
self._body_regex: Optional[re.Pattern] = re.compile(body_regex) if body_regex else None
self._json_path: str | None = json_path
self._body_regex: re.Pattern | None = re.compile(body_regex) if body_regex else None
self._true_values: list[Any] = true_value if isinstance(true_value, list) else [true_value]
self._false_values: list[Any] = false_value if isinstance(false_value, list) else [false_value]

Expand Down Expand Up @@ -92,8 +91,8 @@ async def read_value(self) -> NullablePortValue:
elif isinstance(raw_value, str):
raw_value = raw_value.strip()
factor = 1
if raw_value.endswith('%'):
raw_value = raw_value.strip('%')
if raw_value.endswith("%"):
raw_value = raw_value.strip("%")
factor = 0.01

try:
Expand All @@ -109,4 +108,4 @@ async def read_value(self) -> NullablePortValue:
return None

async def write_value(self, value: NullablePortValue) -> None:
await self.get_peripheral().write_port_value(self, self._write_details, context={'new_value': value})
await self.get_peripheral().write_port_value(self, self._write_details, context={"new_value": value})
19 changes: 0 additions & 19 deletions setup.py

This file was deleted.