Compare commits
152 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| e26e1077e0 | |||
| aa739c30c6 | |||
| 0a321c66fe | |||
| c112f46f01 | |||
| a4f764e178 | |||
| 59f5ee5452 | |||
| f8b057b990 | |||
| e2ba8b144a | |||
| 8d320abe00 | |||
| b77e346736 | |||
| 9fa6b661eb | |||
| 0d1c2735d8 | |||
| 8f3a683851 | |||
| aa8c7b6f42 | |||
| 68036ee130 | |||
| 447f28c9f1 | |||
| 1e7291277e | |||
| 46773aa9d3 | |||
| 480ea91ebb | |||
| d1adccba00 | |||
| 9e30250446 | |||
| 3e7d657e2e | |||
| 7484d8441b | |||
| 682afce05f | |||
| 21b27e2f3b | |||
| 69cd9706cf | |||
| 98075fa2cf | |||
| 782f70fb49 | |||
| 52ab27e60f | |||
| 61e7ebeabd | |||
| 3a8121c306 | |||
| 209a95879f | |||
| 4f829d9908 | |||
| ac5345f528 | |||
| 1142481d89 | |||
| b4690f082f | |||
| 81a98ea938 | |||
| 9b387da77a | |||
| d0724c7d06 | |||
| 1414a3611b | |||
| a553c17c43 | |||
| 0be3fac6ab | |||
| 944e70221e | |||
| 21f13c4750 | |||
| be85ddd159 | |||
| 77c538ced9 | |||
| ee55680b03 | |||
| 0728239915 | |||
| 77016e6f0b | |||
| ce39aa5101 | |||
| f32196ce1b | |||
| 6b24e187a5 | |||
| 129d25df0e | |||
| fa31bd0223 | |||
| 21209b384d | |||
| 1a9c95871f | |||
| dc4d583121 | |||
| ed39f0288f | |||
| 3830706eb1 | |||
| f7ae62ade2 | |||
| 38ffbc27b7 | |||
| 8cebccf250 | |||
| 17aac0b552 | |||
| c281a329a4 | |||
| ca472716db | |||
| af50afe3ff | |||
| b6493df77f | |||
| 59d3c8c3ea | |||
| 4e3405f1fb | |||
| 3772c10b31 | |||
| 242be2fa60 | |||
| 49eabdd712 | |||
| 96a31f0678 | |||
| 91506b0b20 | |||
| b0de9e31b5 | |||
| 2075783134 | |||
| 071f2449c3 | |||
| ffa4f29200 | |||
| 40a691b098 | |||
| 487ce3aeb4 | |||
| 6c0f10ae45 | |||
| 436b26c91c | |||
| f7bac26aed | |||
| a31c049b51 | |||
| a419664461 | |||
| 4a0c07009b | |||
| 682bcbfa9f | |||
| ccf284e8fa | |||
| 23102a28ff | |||
| 5475edb253 | |||
| c52f80c1ef | |||
| 21eecfa24c | |||
| 5dde1bfcf1 | |||
| 82c9d874c9 | |||
| 9acf2c8a92 | |||
| 95012e60c1 | |||
| 19b6500bbd | |||
| 47a06e4630 | |||
| e5a8b40bb2 | |||
| 219456f5f8 | |||
| d1544ae89f | |||
| 8f7d4b2ca7 | |||
| a7d31d4983 | |||
| a89b12a02c | |||
| 15ae68f5d7 | |||
| 0709cd99b5 | |||
| faf06f7141 | |||
| b54e09f811 | |||
| 55b7e0d732 | |||
| 45c922679b | |||
| b1c149382a | |||
| 393e98c8c2 | |||
| 8376329cbb | |||
| 1567fe9e68 | |||
| 364b826a1b | |||
| 297dbab479 | |||
| 81680ed766 | |||
| c934720bb0 | |||
| 9297a5df49 | |||
| 7b8bf49769 | |||
| c834496b72 | |||
| 1c0d6ce8f4 | |||
| 1431e306b8 | |||
| 83bca13c8b | |||
| 1bcef9faf6 | |||
| 8d3e511d18 | |||
| 162f999100 | |||
| 2765c4fa69 | |||
| 69cb2c79c7 | |||
| e2daad36e9 | |||
| d6b06298ed | |||
| 7ddd827340 | |||
| 2a30278e04 | |||
| ecd9089e29 | |||
| bcecaef380 | |||
| 8e986e59aa | |||
| 9e110a125b | |||
| ec8635401b | |||
| f7b867c219 | |||
| e69310619e | |||
| 4a924593b3 | |||
| 1f57e81ddc | |||
| e42a3d4147 | |||
| d3d53ef6a5 | |||
| acb7156bf2 | |||
| 42cda6a477 | |||
| c1dfe5f11f | |||
| 7e57b4cfb6 | |||
| b87237b88f | |||
| fb797e64cb | |||
| 040a49baea | |||
| 105ae0316c |
+19
-9
@@ -1,15 +1,16 @@
|
|||||||
version: 2.1
|
version: 2.1
|
||||||
|
|
||||||
orbs:
|
orbs:
|
||||||
python: circleci/python@1.3.2
|
python: circleci/python@2.0.3
|
||||||
|
codecov: codecov/codecov@3.2.2
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
test:
|
test:
|
||||||
docker:
|
docker:
|
||||||
- image: "cimg/python:<<parameters.tag>>"
|
- image: "cimg/python:<<parameters.tag>>"
|
||||||
- image: "circleci/node:12"
|
- image: "cimg/node:16.15"
|
||||||
- image: "circleci/redis:6"
|
- image: "cimg/redis:6.2"
|
||||||
- image: "circleci/postgres:12"
|
- image: "cimg/postgres:14.2"
|
||||||
environment:
|
environment:
|
||||||
POSTGRES_DB: speckle2_test
|
POSTGRES_DB: speckle2_test
|
||||||
POSTGRES_PASSWORD: speckle
|
POSTGRES_PASSWORD: speckle
|
||||||
@@ -26,6 +27,7 @@ jobs:
|
|||||||
STRATEGY_LOCAL: "true"
|
STRATEGY_LOCAL: "true"
|
||||||
CANONICAL_URL: "http://localhost:3000"
|
CANONICAL_URL: "http://localhost:3000"
|
||||||
WAIT_HOSTS: localhost:5432, localhost:6379
|
WAIT_HOSTS: localhost:5432, localhost:6379
|
||||||
|
DISABLE_FILE_UPLOADS: "true"
|
||||||
parameters:
|
parameters:
|
||||||
tag:
|
tag:
|
||||||
default: "3.8"
|
default: "3.8"
|
||||||
@@ -38,11 +40,19 @@ jobs:
|
|||||||
name: upgrade pip
|
name: upgrade pip
|
||||||
- python/install-packages:
|
- python/install-packages:
|
||||||
pkg-manager: poetry
|
pkg-manager: poetry
|
||||||
- run: poetry run pytest
|
- run: poetry run pytest --cov --cov-report xml:reports/coverage.xml --junitxml=reports/test-results.xml
|
||||||
|
|
||||||
|
- store_test_results:
|
||||||
|
path: reports
|
||||||
|
|
||||||
|
- store_artifacts:
|
||||||
|
path: reports
|
||||||
|
|
||||||
|
- codecov/upload
|
||||||
|
|
||||||
deploy:
|
deploy:
|
||||||
docker:
|
docker:
|
||||||
- image: "circleci/python:3.8"
|
- image: "cimg/python:3.8"
|
||||||
steps:
|
steps:
|
||||||
- checkout
|
- checkout
|
||||||
- run: python patch_version.py $CIRCLE_TAG
|
- run: python patch_version.py $CIRCLE_TAG
|
||||||
@@ -51,17 +61,17 @@ jobs:
|
|||||||
|
|
||||||
workflows:
|
workflows:
|
||||||
main:
|
main:
|
||||||
jobs:
|
jobs:
|
||||||
- test:
|
- test:
|
||||||
matrix:
|
matrix:
|
||||||
parameters:
|
parameters:
|
||||||
tag: ["3.6", "3.7", "3.8", "3.9"]
|
tag: ["3.7", "3.8", "3.9", "3.10"]
|
||||||
filters:
|
filters:
|
||||||
tags:
|
tags:
|
||||||
only: /.*/
|
only: /.*/
|
||||||
- deploy:
|
- deploy:
|
||||||
requires:
|
requires:
|
||||||
- test
|
- test
|
||||||
filters:
|
filters:
|
||||||
tags:
|
tags:
|
||||||
only: /[0-9]+(\.[0-9]+)*/
|
only: /[0-9]+(\.[0-9]+)*/
|
||||||
|
|||||||
@@ -0,0 +1,27 @@
|
|||||||
|
# See here for image contents: https://github.com/microsoft/vscode-dev-containers/blob/main/containers/python-3/.devcontainer/base.Dockerfile
|
||||||
|
|
||||||
|
# [Choice] Python version: 3, 3.9, 3.8, 3.7, 3.6
|
||||||
|
ARG VARIANT="3.10"
|
||||||
|
FROM mcr.microsoft.com/vscode/devcontainers/python:${VARIANT}
|
||||||
|
|
||||||
|
# [Choice] Node.js version: none, lts/*, 16, 14, 12, 10
|
||||||
|
ARG NODE_VERSION="16"
|
||||||
|
RUN if [ "${NODE_VERSION}" != "none" ]; then su vscode -c "umask 0002 && . /usr/local/share/nvm/nvm.sh && nvm install ${NODE_VERSION} 2>&1"; fi
|
||||||
|
|
||||||
|
# [Optional] If your pip requirements rarely change, uncomment this section to add them to the image.
|
||||||
|
# COPY requirements.txt /tmp/pip-tmp/
|
||||||
|
# RUN pip3 --disable-pip-version-check --no-cache-dir install -r /tmp/pip-tmp/requirements.txt \
|
||||||
|
# && rm -rf /tmp/pip-tmp
|
||||||
|
|
||||||
|
# [Optional] Uncomment this section to install additional OS packages.
|
||||||
|
# RUN apt-get update && export DEBIAN_FRONTEND=noninteractive \
|
||||||
|
# && apt-get -y install --no-install-recommends <your-package-list-here>
|
||||||
|
|
||||||
|
# [Optional] Uncomment this line to install global node packages.
|
||||||
|
# RUN su vscode -c "source /usr/local/share/nvm/nvm.sh && npm install -g <your-package-here>" 2>&1
|
||||||
|
|
||||||
|
USER vscode
|
||||||
|
|
||||||
|
RUN curl -sSL https://raw.githubusercontent.com/python-poetry/poetry/master/get-poetry.py | python -
|
||||||
|
|
||||||
|
ENV PATH=$PATH:$HOME/.poetry/env
|
||||||
@@ -0,0 +1,55 @@
|
|||||||
|
// For format details, see https://aka.ms/devcontainer.json. For config options, see the README at:
|
||||||
|
// https://github.com/microsoft/vscode-dev-containers/tree/v0.191.1/containers/python-3
|
||||||
|
{
|
||||||
|
"name": "Python 3",
|
||||||
|
// "build": {
|
||||||
|
// "dockerfile": "Dockerfile",
|
||||||
|
// "context": "..",
|
||||||
|
// "args": {
|
||||||
|
// // Update 'VARIANT' to pick a Python version: 3, 3.6, 3.7, 3.8, 3.9
|
||||||
|
// "VARIANT": "3.6",
|
||||||
|
// // Options
|
||||||
|
// "NODE_VERSION": "lts/*"
|
||||||
|
// }
|
||||||
|
// },
|
||||||
|
"dockerComposeFile": "./docker-compose.yaml",
|
||||||
|
"service": "specklepy",
|
||||||
|
"workspaceFolder": "/workspaces/specklepy",
|
||||||
|
"shutdownAction": "stopCompose",
|
||||||
|
// Set *default* container specific settings.json values on container create.
|
||||||
|
"settings": {
|
||||||
|
"python.pythonPath": "/usr/local/bin/python",
|
||||||
|
"python.languageServer": "Pylance",
|
||||||
|
"python.linting.enabled": true,
|
||||||
|
"python.linting.pylintEnabled": true,
|
||||||
|
"python.linting.pylintArgs": [
|
||||||
|
"--max-line-length=120"
|
||||||
|
],
|
||||||
|
"python.formatting.autopep8Path": "/usr/local/py-utils/bin/autopep8",
|
||||||
|
"python.formatting.blackPath": "/usr/local/py-utils/bin/black",
|
||||||
|
"python.formatting.yapfPath": "/usr/local/py-utils/bin/yapf",
|
||||||
|
"python.linting.banditPath": "/usr/local/py-utils/bin/bandit",
|
||||||
|
"python.linting.flake8Path": "/usr/local/py-utils/bin/flake8",
|
||||||
|
"python.linting.mypyPath": "/usr/local/py-utils/bin/mypy",
|
||||||
|
"python.linting.pycodestylePath": "/usr/local/py-utils/bin/pycodestyle",
|
||||||
|
"python.linting.pydocstylePath": "/usr/local/py-utils/bin/pydocstyle",
|
||||||
|
"python.linting.pylintPath": "/usr/local/py-utils/bin/pylint",
|
||||||
|
"python.testing.pytestArgs": [
|
||||||
|
"tests/",
|
||||||
|
"-s"
|
||||||
|
],
|
||||||
|
"python.testing.pytestEnabled": true,
|
||||||
|
"editor.formatOnSave": true,
|
||||||
|
},
|
||||||
|
// Add the IDs of extensions you want installed when the container is created.
|
||||||
|
"extensions": [
|
||||||
|
"ms-python.python",
|
||||||
|
"ms-python.vscode-pylance"
|
||||||
|
],
|
||||||
|
// Use 'forwardPorts' to make a list of ports inside the container available locally.
|
||||||
|
// "forwardPorts": [],
|
||||||
|
// Use 'postCreateCommand' to run commands after the container is created.
|
||||||
|
"postCreateCommand": "poetry config virtualenvs.create false && poetry install",
|
||||||
|
// Comment out connect as root instead. More info: https://aka.ms/vscode-remote/containers/non-root.
|
||||||
|
"remoteUser": "vscode"
|
||||||
|
}
|
||||||
@@ -0,0 +1,44 @@
|
|||||||
|
version: "3.3" # optional since v1.27.0
|
||||||
|
services:
|
||||||
|
postgres:
|
||||||
|
image: cimg/postgres:14.2
|
||||||
|
environment:
|
||||||
|
POSTGRES_DB: speckle2_test
|
||||||
|
POSTGRES_PASSWORD: speckle
|
||||||
|
POSTGRES_USER: speckle
|
||||||
|
network_mode: host
|
||||||
|
redis:
|
||||||
|
image: cimg/redis:6.2
|
||||||
|
network_mode: host
|
||||||
|
speckle-server:
|
||||||
|
image: speckle/speckle-server:latest
|
||||||
|
command: ["bash", "-c", "/wait && node bin/www"]
|
||||||
|
environment:
|
||||||
|
POSTGRES_URL: "localhost"
|
||||||
|
POSTGRES_USER: "speckle"
|
||||||
|
POSTGRES_PASSWORD: "speckle"
|
||||||
|
POSTGRES_DB: "speckle2_test"
|
||||||
|
REDIS_URL: "redis://localhost"
|
||||||
|
SESSION_SECRET: "keyboard cat"
|
||||||
|
STRATEGY_LOCAL: "true"
|
||||||
|
CANONICAL_URL: "http://localhost:3000"
|
||||||
|
WAIT_HOSTS: localhost:5432, localhost:6379
|
||||||
|
DISABLE_FILE_UPLOADS: "true"
|
||||||
|
network_mode: host
|
||||||
|
|
||||||
|
specklepy:
|
||||||
|
build:
|
||||||
|
dockerfile: Dockerfile
|
||||||
|
context: .
|
||||||
|
args:
|
||||||
|
VARIANT: 3.9
|
||||||
|
NODE_VERSION: lts/*
|
||||||
|
volumes:
|
||||||
|
# Mounts the project folder to '/workspace'. While this file is in .devcontainer,
|
||||||
|
# mounts are relative to the first file in the list, which is a level up.
|
||||||
|
- ..:/workspaces/specklepy:cached
|
||||||
|
# Overrides default command so things don't shut down after the process ends.
|
||||||
|
command: /bin/sh -c "while sleep 1000; do :; done"
|
||||||
|
network_mode: host
|
||||||
|
# networks:
|
||||||
|
# default:
|
||||||
@@ -0,0 +1,3 @@
|
|||||||
|
* text=auto eol=lf
|
||||||
|
*.{cmd,[cC][mM][dD]} text eol=crlf
|
||||||
|
*.{bat,[bB][aA][tT]} text eol=crlf
|
||||||
@@ -1,25 +0,0 @@
|
|||||||
---
|
|
||||||
name: New issue
|
|
||||||
about: Create a report to help us improve
|
|
||||||
title:
|
|
||||||
labels:
|
|
||||||
assignees:
|
|
||||||
---
|
|
||||||
|
|
||||||
If it's your first time here - or you forgot about them - make sure you read the [contribution guidelines](CONTRIBUTING.md), and then feel free to delete this line!
|
|
||||||
|
|
||||||
### Expected vs. Actual Behavior
|
|
||||||
|
|
||||||
Describe the problem here.
|
|
||||||
|
|
||||||
### Reproduction Steps & System Config (win, osx, web, etc.)
|
|
||||||
|
|
||||||
Let us know how we can reproduce this, and attach relevant files (if any).
|
|
||||||
|
|
||||||
### Proposed Solution (if any)
|
|
||||||
|
|
||||||
Let us know what how you would solve this.
|
|
||||||
|
|
||||||
#### Optional: Affected Projects
|
|
||||||
|
|
||||||
Does this issue propagate to other dependencies or dependents? If so, list them here!
|
|
||||||
@@ -0,0 +1,113 @@
|
|||||||
|
---
|
||||||
|
name: Bug report
|
||||||
|
about: Help improve Speckle!
|
||||||
|
title: ''
|
||||||
|
labels: bug
|
||||||
|
assignees: ''
|
||||||
|
---
|
||||||
|
|
||||||
|
<!---
|
||||||
|
|
||||||
|
Provide a short summary in the Title above. Examples of good Issue titles:
|
||||||
|
|
||||||
|
* "Bug: Error from server when reticulating splines"
|
||||||
|
* "Bug: Revit crashes when installing connector"
|
||||||
|
|
||||||
|
-->
|
||||||
|
|
||||||
|
## Prerequisites
|
||||||
|
|
||||||
|
<!---
|
||||||
|
|
||||||
|
Please answer the following questions before submitting an issue.
|
||||||
|
|
||||||
|
-->
|
||||||
|
|
||||||
|
- [ ] I read the [contribution guidelines](https://github.com/specklesystems/speckle-server/blob/main/CONTRIBUTING.md)
|
||||||
|
- [ ] I checked the [documentation](https://speckle.guide/) and found no answer.
|
||||||
|
- [ ] I checked [existing issues](../issues?q=is%3Aissue) and found no similar issue. <!-- If you do find an existing issue, please show your support by liking it :+1: instead of creating a new issue -->
|
||||||
|
- [ ] I checked the [community forum](https://speckle.community/) for related discussions and found no answer.
|
||||||
|
- [ ] I'm reporting the issue to the correct repository (see also [speckle-server](https://github.com/specklesystems/speckle-server), [speckle-sharp](https://github.com/specklesystems/speckle-sharp), [specklepy](https://github.com/specklesystems/specklepy), [speckle-docs](https://github.com/specklesystems/speckle-docs), and [others](https://github.com/orgs/specklesystems/repositories))
|
||||||
|
|
||||||
|
## What package are you referring to?
|
||||||
|
|
||||||
|
<!---
|
||||||
|
Is it related to the server (backend) only, or does this bug relate to the frontend, viewer, objectloader or any other package?
|
||||||
|
-->
|
||||||
|
|
||||||
|
## Describe the bug
|
||||||
|
|
||||||
|
<!---
|
||||||
|
A clear and concise description of what the bug is.
|
||||||
|
-->
|
||||||
|
|
||||||
|
## To Reproduce
|
||||||
|
|
||||||
|
<!---
|
||||||
|
Steps to reproduce the behavior:
|
||||||
|
|
||||||
|
1. Go to '...'
|
||||||
|
2. Click on '....'
|
||||||
|
3. Scroll down to '....'
|
||||||
|
4. See error
|
||||||
|
-->
|
||||||
|
|
||||||
|
## Expected behavior
|
||||||
|
|
||||||
|
<!---
|
||||||
|
A clear and concise description of what you expected to happen.
|
||||||
|
-->
|
||||||
|
|
||||||
|
## Screenshots
|
||||||
|
|
||||||
|
<!---
|
||||||
|
If applicable, add screenshots to help explain your problem.
|
||||||
|
-->
|
||||||
|
|
||||||
|
## System Info
|
||||||
|
|
||||||
|
If applicable, please fill in the below details - they help a lot!
|
||||||
|
|
||||||
|
### Desktop (please complete the following information):
|
||||||
|
|
||||||
|
- OS: [e.g. iOS]
|
||||||
|
- Browser [e.g. chrome, safari]
|
||||||
|
- Version [e.g. 22]
|
||||||
|
|
||||||
|
### Smartphone (please complete the following information):
|
||||||
|
|
||||||
|
- Device: [e.g. iPhone6]
|
||||||
|
- OS: [e.g. iOS8.1]
|
||||||
|
- Browser [e.g. stock browser, safari]
|
||||||
|
- Version [e.g. 22]
|
||||||
|
|
||||||
|
## Failure Logs
|
||||||
|
|
||||||
|
<!---
|
||||||
|
Please include any relevant log snippets or files here, or upload as a file.
|
||||||
|
|
||||||
|
If including inline, please use markdown code block syntax. https://docs.github.com/en/get-started/writing-on-github/working-with-advanced-formatting/creating-and-highlighting-code-blocks
|
||||||
|
For example:
|
||||||
|
|
||||||
|
```
|
||||||
|
your log output here
|
||||||
|
```
|
||||||
|
-->
|
||||||
|
|
||||||
|
## Additional context
|
||||||
|
|
||||||
|
<!---
|
||||||
|
Add any other context about the problem here.
|
||||||
|
-->
|
||||||
|
|
||||||
|
## Proposed Solution (if any)
|
||||||
|
|
||||||
|
<!---
|
||||||
|
Let us know what how you would solve this.
|
||||||
|
-->
|
||||||
|
|
||||||
|
#### Optional: Affected Projects
|
||||||
|
|
||||||
|
<!---
|
||||||
|
Does this issue propagate to other dependencies or dependents? If so, list them here with links!
|
||||||
|
-->
|
||||||
@@ -0,0 +1,71 @@
|
|||||||
|
---
|
||||||
|
name: Feature request
|
||||||
|
about: Suggest an idea for Speckle!
|
||||||
|
title: ''
|
||||||
|
labels: enhancement, question
|
||||||
|
assignees: ''
|
||||||
|
---
|
||||||
|
|
||||||
|
<!---
|
||||||
|
|
||||||
|
Provide a short summary in the Title above. Examples of good Issue titles:
|
||||||
|
|
||||||
|
* "Enhancement: Connector for Minecraft"
|
||||||
|
* "Enhancement: Web viewer should support tesseracts"
|
||||||
|
|
||||||
|
-->
|
||||||
|
|
||||||
|
## Prerequisites
|
||||||
|
|
||||||
|
<!---
|
||||||
|
|
||||||
|
Please answer the following questions before submitting an issue.
|
||||||
|
|
||||||
|
-->
|
||||||
|
|
||||||
|
- [ ] I read the [contribution guidelines](https://github.com/specklesystems/speckle-server/blob/main/CONTRIBUTING.md)
|
||||||
|
- [ ] I checked the [documentation](https://speckle.guide/) and found no answer.
|
||||||
|
- [ ] I checked [existing issues](../issues?q=is%3Aissue) and found no similar issue. <!-- If you do find an existing issue, please show your support by liking it :+1: instead of creating a new issue -->
|
||||||
|
- [ ] I checked the [community forum](https://speckle.community/) for related discussions and found no answer.
|
||||||
|
- [ ] I'm requesting the feature to the correct repository (see also [speckle-server](https://github.com/specklesystems/speckle-server), [speckle-sharp](https://github.com/specklesystems/speckle-sharp), [specklepy](https://github.com/specklesystems/specklepy), [speckle-docs](https://github.com/specklesystems/speckle-docs), and [others](https://github.com/orgs/specklesystems/repositories))
|
||||||
|
|
||||||
|
## What package are you referring to?
|
||||||
|
|
||||||
|
<!---
|
||||||
|
Is it related to the server (backend) only, or does this feature request relate to the frontend, viewer, objectloader or any other package?
|
||||||
|
-->
|
||||||
|
|
||||||
|
## Is your feature request related to a problem? Please describe.
|
||||||
|
|
||||||
|
<!---
|
||||||
|
A clear and concise description of what the problem is. Ex. I'm always frustrated when [...]
|
||||||
|
-->
|
||||||
|
|
||||||
|
## Describe the solution you'd like
|
||||||
|
|
||||||
|
<!---
|
||||||
|
A clear and concise description of what you want to happen.
|
||||||
|
-->
|
||||||
|
|
||||||
|
## Describe alternatives you've considered
|
||||||
|
|
||||||
|
<!---
|
||||||
|
A clear and concise description of any alternative solutions or features you've considered.
|
||||||
|
-->
|
||||||
|
|
||||||
|
## Additional context
|
||||||
|
|
||||||
|
<!---
|
||||||
|
Add any other context or screenshots about the feature request here.
|
||||||
|
|
||||||
|
Have you seen this feature implemented in any other software? Can you provide screenshots or links to video or documentation?
|
||||||
|
What works well about these existing features in other software? What doesn't work well?
|
||||||
|
-->
|
||||||
|
|
||||||
|
## Related issues or community discussions
|
||||||
|
|
||||||
|
<!---
|
||||||
|
Is this feature request related to (but sufficiently distinct from) any existing issues?
|
||||||
|
Does this feature request require other features to be available beforehand?
|
||||||
|
Has this feature been discussed in the community forum, please link here? https://speckle.community/
|
||||||
|
-->
|
||||||
@@ -1,25 +0,0 @@
|
|||||||
Description of PR...
|
|
||||||
|
|
||||||
## Changes
|
|
||||||
|
|
||||||
- Item 1
|
|
||||||
- Item 2
|
|
||||||
|
|
||||||
## Checklist
|
|
||||||
|
|
||||||
- [ ] Unit tests
|
|
||||||
- [ ] Documentation
|
|
||||||
|
|
||||||
## References
|
|
||||||
|
|
||||||
(optional)
|
|
||||||
|
|
||||||
Include **important** links regarding the implementation of this PR.
|
|
||||||
This usually includes and RFC or an aggregation of issues and/or individual conversations
|
|
||||||
that helped put this solution together. This helps ensure there is a good aggregation
|
|
||||||
of resources regarding the implementation.
|
|
||||||
|
|
||||||
```text
|
|
||||||
Fixes #85, Fixes #22, Fixes username/repo#123
|
|
||||||
Connects #123
|
|
||||||
```
|
|
||||||
@@ -0,0 +1,102 @@
|
|||||||
|
<!---
|
||||||
|
|
||||||
|
Provide a short summary in the Title above. Examples of good PR titles:
|
||||||
|
|
||||||
|
* "Feature: adds metrics to component"
|
||||||
|
|
||||||
|
* "Fix: resolves duplication in comment thread"
|
||||||
|
|
||||||
|
* "Update: apollo v2.34.0"
|
||||||
|
|
||||||
|
-->
|
||||||
|
|
||||||
|
## Description & motivation
|
||||||
|
|
||||||
|
<!---
|
||||||
|
|
||||||
|
Describe your changes, and why you're making them. What benefit will this have to others?
|
||||||
|
|
||||||
|
Is this linked to an open Github issue, a thread in Speckle community,
|
||||||
|
or another pull request? Link it here.
|
||||||
|
|
||||||
|
If it is related to a Github issue, and resolves it, please link to the issue number, e.g.:
|
||||||
|
Fixes #85, Fixes #22, Fixes username/repo#123
|
||||||
|
Connects #123
|
||||||
|
|
||||||
|
-->
|
||||||
|
|
||||||
|
## Changes:
|
||||||
|
|
||||||
|
<!---
|
||||||
|
|
||||||
|
- Item 1
|
||||||
|
- Item 2
|
||||||
|
|
||||||
|
-->
|
||||||
|
|
||||||
|
## To-do before merge:
|
||||||
|
|
||||||
|
<!---
|
||||||
|
|
||||||
|
(Optional -- remove this section if not needed)
|
||||||
|
|
||||||
|
Include any notes about things that need to happen before this PR is merged, e.g.:
|
||||||
|
|
||||||
|
- [ ] Change the base branch
|
||||||
|
|
||||||
|
- [ ] Ensure PR #56 is merged
|
||||||
|
|
||||||
|
-->
|
||||||
|
|
||||||
|
## Screenshots:
|
||||||
|
|
||||||
|
<!---
|
||||||
|
|
||||||
|
Include a screenshot the before and after. This can be a screenshot of a plugin, web frontend, or output in a terminal.
|
||||||
|
|
||||||
|
-->
|
||||||
|
|
||||||
|
## Validation of changes:
|
||||||
|
|
||||||
|
<!---
|
||||||
|
|
||||||
|
Describe what tests have been added or amended, and why these demonstrate it works and will prevent this feature being accidentally broken by future changes.
|
||||||
|
|
||||||
|
-->
|
||||||
|
|
||||||
|
## Checklist:
|
||||||
|
|
||||||
|
<!---
|
||||||
|
|
||||||
|
This checklist is mostly useful as a reminder of small things that can easily be
|
||||||
|
|
||||||
|
forgotten – it is meant as a helpful tool rather than hoops to jump through.
|
||||||
|
|
||||||
|
Put an `x` between the square brackets, e.g. [x], for all the items that apply,
|
||||||
|
|
||||||
|
make notes next to any that haven't been addressed, and remove any items that are not relevant to this PR.
|
||||||
|
|
||||||
|
-->
|
||||||
|
|
||||||
|
- [ ] My pull request follows the guidelines in the [Contributing guide](https://github.com/specklesystems/speckle-server/blob/main/CONTRIBUTING.md)?
|
||||||
|
- [ ] My pull request does not duplicate any other open [Pull Requests](../../pulls) for the same update/change?
|
||||||
|
- [ ] My commits are related to the pull request and do not amend unrelated code or documentation.
|
||||||
|
- [ ] My code follows a similar style to existing code.
|
||||||
|
- [ ] I have added appropriate tests.
|
||||||
|
- [ ] I have updated or added relevant documentation.
|
||||||
|
|
||||||
|
## References
|
||||||
|
|
||||||
|
<!---
|
||||||
|
|
||||||
|
(Optional -- remove this section if not needed )
|
||||||
|
|
||||||
|
Include **important** links regarding the implementation of this PR.
|
||||||
|
|
||||||
|
This usually includes a RFC or an aggregation of issues and/or individual conversations
|
||||||
|
|
||||||
|
that helped put this solution together. This helps ensure we retain and share knowledge
|
||||||
|
|
||||||
|
regarding the implementation, and may help others understand motivation and design decisions etc..
|
||||||
|
|
||||||
|
-->
|
||||||
@@ -32,10 +32,10 @@ jobs:
|
|||||||
|
|
||||||
echo 'PROJECT_ID='$(jq '.data.organization.projectNext.id' project_data.json) >> $GITHUB_ENV
|
echo 'PROJECT_ID='$(jq '.data.organization.projectNext.id' project_data.json) >> $GITHUB_ENV
|
||||||
echo 'STATUS_FIELD_ID='$(jq '.data.organization.projectNext.fields.nodes[] | select(.name== "Status") | .id' project_data.json) >> $GITHUB_ENV
|
echo 'STATUS_FIELD_ID='$(jq '.data.organization.projectNext.fields.nodes[] | select(.name== "Status") | .id' project_data.json) >> $GITHUB_ENV
|
||||||
|
|
||||||
echo "$PROJECT_ID"
|
echo "$PROJECT_ID"
|
||||||
echo "$STATUS_FIELD_ID"
|
echo "$STATUS_FIELD_ID"
|
||||||
|
|
||||||
echo 'DONE_ID='$(jq '.data.organization.projectNext.fields.nodes[] | select(.name== "Status") | .settings | fromjson | .options[] | select(.name== "Done") | .id' project_data.json) >> $GITHUB_ENV
|
echo 'DONE_ID='$(jq '.data.organization.projectNext.fields.nodes[] | select(.name== "Status") | .settings | fromjson | .options[] | select(.name== "Done") | .id' project_data.json) >> $GITHUB_ENV
|
||||||
echo "$DONE_ID"
|
echo "$DONE_ID"
|
||||||
|
|
||||||
@@ -52,9 +52,9 @@ jobs:
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}' -f project=$PROJECT_ID -f id=$ISSUE_ID --jq '.data.addProjectNextItem.projectNextItem.id')"
|
}' -f project=$PROJECT_ID -f id=$ISSUE_ID --jq '.data.addProjectNextItem.projectNextItem.id')"
|
||||||
|
|
||||||
echo 'ITEM_ID='$item_id >> $GITHUB_ENV
|
echo 'ITEM_ID='$item_id >> $GITHUB_ENV
|
||||||
|
|
||||||
- name: Update Status
|
- name: Update Status
|
||||||
env:
|
env:
|
||||||
GITHUB_TOKEN: ${{secrets.GHPROJECT_TOKEN}}
|
GITHUB_TOKEN: ${{secrets.GHPROJECT_TOKEN}}
|
||||||
@@ -75,4 +75,3 @@ jobs:
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}' -f project=$PROJECT_ID -f status=$STATUS_FIELD_ID -f id=$ITEM_ID -f value=${{ env.DONE_ID }}
|
}' -f project=$PROJECT_ID -f status=$STATUS_FIELD_ID -f id=$ITEM_ID -f value=${{ env.DONE_ID }}
|
||||||
|
|
||||||
|
|||||||
@@ -32,7 +32,7 @@ jobs:
|
|||||||
|
|
||||||
echo 'PROJECT_ID='$(jq '.data.organization.projectNext.id' project_data.json) >> $GITHUB_ENV
|
echo 'PROJECT_ID='$(jq '.data.organization.projectNext.id' project_data.json) >> $GITHUB_ENV
|
||||||
echo 'STATUS_FIELD_ID='$(jq '.data.organization.projectNext.fields.nodes[] | select(.name== "Status") | .id' project_data.json) >> $GITHUB_ENV
|
echo 'STATUS_FIELD_ID='$(jq '.data.organization.projectNext.fields.nodes[] | select(.name== "Status") | .id' project_data.json) >> $GITHUB_ENV
|
||||||
|
|
||||||
- name: Add Issue to project
|
- name: Add Issue to project
|
||||||
env:
|
env:
|
||||||
GITHUB_TOKEN: ${{secrets.GHPROJECT_TOKEN}}
|
GITHUB_TOKEN: ${{secrets.GHPROJECT_TOKEN}}
|
||||||
@@ -46,5 +46,5 @@ jobs:
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}' -f project=$PROJECT_ID -f id=$ISSUE_ID --jq '.data.addProjectNextItem.projectNextItem.id')"
|
}' -f project=$PROJECT_ID -f id=$ISSUE_ID --jq '.data.addProjectNextItem.projectNextItem.id')"
|
||||||
|
|
||||||
echo 'ITEM_ID='$item_id >> $GITHUB_ENV
|
echo 'ITEM_ID='$item_id >> $GITHUB_ENV
|
||||||
|
|||||||
@@ -1,3 +1,7 @@
|
|||||||
|
.tool-versions
|
||||||
|
.envrc
|
||||||
|
reports/
|
||||||
|
|
||||||
# Byte-compiled / optimized / DLL files
|
# Byte-compiled / optimized / DLL files
|
||||||
__pycache__/
|
__pycache__/
|
||||||
*.py[cod]
|
*.py[cod]
|
||||||
|
|||||||
Vendored
+6
-2
@@ -4,6 +4,7 @@
|
|||||||
// For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
|
// For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
|
||||||
"version": "0.2.0",
|
"version": "0.2.0",
|
||||||
"configurations": [
|
"configurations": [
|
||||||
|
|
||||||
{
|
{
|
||||||
"name": "Python: Current File",
|
"name": "Python: Current File",
|
||||||
"type": "python",
|
"type": "python",
|
||||||
@@ -13,10 +14,13 @@
|
|||||||
"justMyCode": false
|
"justMyCode": false
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
"name": "Python: Test debug config",
|
"name": "Pytest",
|
||||||
"type": "python",
|
"type": "python",
|
||||||
"request": "test",
|
"request": "launch",
|
||||||
|
"program": "poetry",
|
||||||
|
"args": ["run", "pytest"],
|
||||||
"console": "integratedTerminal",
|
"console": "integratedTerminal",
|
||||||
|
"justMyCode": true
|
||||||
}
|
}
|
||||||
]
|
]
|
||||||
}
|
}
|
||||||
@@ -8,7 +8,9 @@
|
|||||||
<p align="center"><b>Speckle</b> is the data infrastructure for the AEC industry.</p><br/>
|
<p align="center"><b>Speckle</b> is the data infrastructure for the AEC industry.</p><br/>
|
||||||
|
|
||||||
<p align="center"><a href="https://twitter.com/SpeckleSystems"><img src="https://img.shields.io/twitter/follow/SpeckleSystems?style=social" alt="Twitter Follow"></a> <a href="https://speckle.community"><img src="https://img.shields.io/discourse/users?server=https%3A%2F%2Fspeckle.community&style=flat-square&logo=discourse&logoColor=white" alt="Community forum users"></a> <a href="https://speckle.systems"><img src="https://img.shields.io/badge/https://-speckle.systems-royalblue?style=flat-square" alt="website"></a> <a href="https://speckle.guide/dev/"><img src="https://img.shields.io/badge/docs-speckle.guide-orange?style=flat-square&logo=read-the-docs&logoColor=white" alt="docs"></a></p>
|
<p align="center"><a href="https://twitter.com/SpeckleSystems"><img src="https://img.shields.io/twitter/follow/SpeckleSystems?style=social" alt="Twitter Follow"></a> <a href="https://speckle.community"><img src="https://img.shields.io/discourse/users?server=https%3A%2F%2Fspeckle.community&style=flat-square&logo=discourse&logoColor=white" alt="Community forum users"></a> <a href="https://speckle.systems"><img src="https://img.shields.io/badge/https://-speckle.systems-royalblue?style=flat-square" alt="website"></a> <a href="https://speckle.guide/dev/"><img src="https://img.shields.io/badge/docs-speckle.guide-orange?style=flat-square&logo=read-the-docs&logoColor=white" alt="docs"></a></p>
|
||||||
<p align="center"><a href="https://github.com/specklesystems/specklepy/"><img src="https://circleci.com/gh/specklesystems/specklepy.svg?style=svg&circle-token=76eabd350ea243575cbb258b746ed3f471f7ac29" alt="Speckle-Next"></a> </p>
|
<p align="center"><a href="https://github.com/specklesystems/specklepy/"><img src="https://circleci.com/gh/specklesystems/specklepy.svg?style=svg&circle-token=76eabd350ea243575cbb258b746ed3f471f7ac29" alt="Speckle-Next"></a><a href="https://codecov.io/gh/specklesystems/specklepy">
|
||||||
|
<img src="https://codecov.io/gh/specklesystems/specklepy/branch/main/graph/badge.svg?token=8KQFL5N0YF"/>
|
||||||
|
</a> </p>
|
||||||
|
|
||||||
# About Speckle
|
# About Speckle
|
||||||
|
|
||||||
@@ -72,7 +74,7 @@ It may be helpful to know where the local accounts and object cache dbs are stor
|
|||||||
|
|
||||||
## Contributing
|
## Contributing
|
||||||
|
|
||||||
Please make sure you read the [contribution guidelines](.github/CONTRIBUTING.md) for an overview of the best practices we try to follow.
|
Please make sure you read the [contribution guidelines](.github/CONTRIBUTING.md) and [code of conduct](.github/CODE_OF_CONDUCT.md) for an overview of the practices we try to follow.
|
||||||
|
|
||||||
## Community
|
## Community
|
||||||
|
|
||||||
|
|||||||
+12
@@ -0,0 +1,12 @@
|
|||||||
|
# Security Policy
|
||||||
|
|
||||||
|
## Supported Versions
|
||||||
|
|
||||||
|
| Version | Supported |
|
||||||
|
| ------- | ------------------ |
|
||||||
|
| 2.2.+ | :white_check_mark: |
|
||||||
|
| < 2.2 | :x: |
|
||||||
|
|
||||||
|
## Reporting a Vulnerability
|
||||||
|
|
||||||
|
Hi! If you've found something off, we'd be more than happy if you would report it via security@speckle.systems. We will work together with you to correctly identify the cause and implement a fix. Thanks for helping make Speckle safer!
|
||||||
@@ -0,0 +1,63 @@
|
|||||||
|
from typing import List
|
||||||
|
from specklepy.objects import Base
|
||||||
|
from specklepy.api import operations
|
||||||
|
from specklepy.transports.sqlite import SQLiteTransport
|
||||||
|
import time
|
||||||
|
from pathlib import Path
|
||||||
|
import os
|
||||||
|
import string
|
||||||
|
import random
|
||||||
|
|
||||||
|
|
||||||
|
class Sub(Base):
|
||||||
|
bar: List[str]
|
||||||
|
|
||||||
|
|
||||||
|
def random_string():
|
||||||
|
letters = string.ascii_lowercase
|
||||||
|
return "".join(random.choice(letters) for _ in range(10))
|
||||||
|
|
||||||
|
|
||||||
|
BASE_PATH = SQLiteTransport.get_base_path("Speckle")
|
||||||
|
|
||||||
|
|
||||||
|
def clean_db():
|
||||||
|
os.remove(Path(BASE_PATH, "Objects.db"))
|
||||||
|
|
||||||
|
|
||||||
|
def one_pass(clean: bool, randomize: bool, child_count: int):
|
||||||
|
|
||||||
|
foo = Base()
|
||||||
|
for i in range(child_count):
|
||||||
|
stuff = random_string() if randomize else "stuff"
|
||||||
|
foo[f"@child_{i}"] = Sub(bar=["asdf", "bar", i, stuff])
|
||||||
|
|
||||||
|
if clean:
|
||||||
|
clean_db()
|
||||||
|
transport = SQLiteTransport()
|
||||||
|
start = time.time()
|
||||||
|
hash = operations.send(base=foo, transports=[transport])
|
||||||
|
send_time = time.time() - start
|
||||||
|
|
||||||
|
receive_start = time.time()
|
||||||
|
operations.receive(hash, transport)
|
||||||
|
receive_time = time.time() - receive_start
|
||||||
|
return send_time, receive_time
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
sample_size = 4
|
||||||
|
|
||||||
|
test_permutations = [
|
||||||
|
(True, True),
|
||||||
|
(False, False),
|
||||||
|
(False, True),
|
||||||
|
(True, False),
|
||||||
|
]
|
||||||
|
for clean, randomize in test_permutations:
|
||||||
|
print(f"CLEAN: {clean}, RANDOMIZE: {randomize}")
|
||||||
|
for child_count in [10, 100, 1000, 10000]:
|
||||||
|
print(f"\tCHILD COUNT: {child_count}")
|
||||||
|
for _ in range(sample_size):
|
||||||
|
send_time, receive_time = one_pass(clean, randomize, child_count)
|
||||||
|
print(f"\t\tSend: {send_time} Receive: {receive_time}")
|
||||||
@@ -50,10 +50,10 @@ if __name__ == "__main__":
|
|||||||
)
|
)
|
||||||
# support for dynamic attributes
|
# support for dynamic attributes
|
||||||
custom_sub.extra_extra = "what is this?"
|
custom_sub.extra_extra = "what is this?"
|
||||||
debug(custom_sub.json())
|
debug(custom_sub)
|
||||||
|
|
||||||
serialized = operations.serialize(custom_sub)
|
serialized = operations.serialize(custom_sub)
|
||||||
deserialized = operations.deserialize(serialized)
|
deserialized = operations.deserialize(serialized)
|
||||||
# the only difference should be between the two data is that the deserialized
|
# the only difference should be between the two data is that the deserialized
|
||||||
# instance id attribute is not None.
|
# instance id attribute is not None.
|
||||||
debug(deserialized.json())
|
debug(deserialized)
|
||||||
|
|||||||
Generated
+858
-442
File diff suppressed because it is too large
Load Diff
+10
-6
@@ -1,6 +1,6 @@
|
|||||||
[tool.poetry]
|
[tool.poetry]
|
||||||
name = "specklepy"
|
name = "specklepy"
|
||||||
version = "2.1.0"
|
version = "2.4.0"
|
||||||
description = "The Python SDK for Speckle 2.0"
|
description = "The Python SDK for Speckle 2.0"
|
||||||
readme = "README.md"
|
readme = "README.md"
|
||||||
authors = ["Speckle Systems <devops@speckle.systems>"]
|
authors = ["Speckle Systems <devops@speckle.systems>"]
|
||||||
@@ -11,17 +11,21 @@ homepage = "https://speckle.systems/"
|
|||||||
|
|
||||||
|
|
||||||
[tool.poetry.dependencies]
|
[tool.poetry.dependencies]
|
||||||
python = "^3.6.5"
|
python = ">=3.7.2, <4.0"
|
||||||
pydantic = "^1.7.3"
|
pydantic = "^1.8.2"
|
||||||
appdirs = "^1.4.4"
|
appdirs = "^1.4.4"
|
||||||
gql = {version = ">=3.0.0a6", extras = ["all"], allow-prereleases = true}
|
gql = {extras = ["requests", "websockets"], version = "^3.3.0"}
|
||||||
ujson = "^4.1.0"
|
ujson = "^5.3.0"
|
||||||
|
Deprecated = "^1.2.13"
|
||||||
|
|
||||||
[tool.poetry.dev-dependencies]
|
[tool.poetry.dev-dependencies]
|
||||||
black = "^20.8b1"
|
black = "^20.8b1"
|
||||||
isort = "^5.7.0"
|
isort = "^5.7.0"
|
||||||
pytest = "^6.2.2"
|
pytest = "^6.2.2"
|
||||||
pytest-ordering = "^0.6"
|
pytest-ordering = "^0.6"
|
||||||
|
pytest-cov = "^3.0.0"
|
||||||
|
devtools = "^0.8.0"
|
||||||
|
pylint = "^2.14.4"
|
||||||
|
|
||||||
|
|
||||||
[tool.black]
|
[tool.black]
|
||||||
@@ -41,7 +45,7 @@ exclude = '''
|
|||||||
'''
|
'''
|
||||||
include = '\.pyi?$'
|
include = '\.pyi?$'
|
||||||
line-length = 88
|
line-length = 88
|
||||||
target-version = ["py36", "py37", "py38"]
|
target-version = ["py37", "py38", "py39", "py310"]
|
||||||
|
|
||||||
|
|
||||||
[build-system]
|
[build-system]
|
||||||
|
|||||||
+91
-31
@@ -1,6 +1,12 @@
|
|||||||
import re
|
import re
|
||||||
from gql.client import SyncClientSession
|
from warnings import warn
|
||||||
from specklepy.logging.exceptions import SpeckleException
|
from deprecated import deprecated
|
||||||
|
from specklepy.api.credentials import Account, get_account_from_token
|
||||||
|
from specklepy.logging import metrics
|
||||||
|
from specklepy.logging.exceptions import (
|
||||||
|
SpeckleException,
|
||||||
|
SpeckleWarning,
|
||||||
|
)
|
||||||
from typing import Dict
|
from typing import Dict
|
||||||
|
|
||||||
from specklepy.api import resources
|
from specklepy.api import resources
|
||||||
@@ -14,9 +20,8 @@ from specklepy.api.resources import (
|
|||||||
subscriptions,
|
subscriptions,
|
||||||
)
|
)
|
||||||
from specklepy.api.models import ServerInfo
|
from specklepy.api.models import ServerInfo
|
||||||
from gql import Client, gql
|
from gql import Client
|
||||||
from gql.transport.requests import RequestsHTTPTransport
|
from gql.transport.requests import RequestsHTTPTransport
|
||||||
from gql.transport.aiohttp import AIOHTTPTransport
|
|
||||||
from gql.transport.websockets import WebsocketsTransport
|
from gql.transport.websockets import WebsocketsTransport
|
||||||
|
|
||||||
|
|
||||||
@@ -36,9 +41,9 @@ class SpeckleClient:
|
|||||||
client = SpeckleClient(host="speckle.xyz") # or whatever your host is
|
client = SpeckleClient(host="speckle.xyz") # or whatever your host is
|
||||||
# client = SpeckleClient(host="localhost:3000", use_ssl=False) or use local server
|
# client = SpeckleClient(host="localhost:3000", use_ssl=False) or use local server
|
||||||
|
|
||||||
# authenticate the client with a token (account has been added in Speckle Manager)
|
# authenticate the client with an account (account has been added in Speckle Manager)
|
||||||
account = get_default_account()
|
account = get_default_account()
|
||||||
client.authenticate(token=account.token)
|
client.authenticate_with_account(account)
|
||||||
|
|
||||||
# create a new stream. this returns the stream id
|
# create a new stream. this returns the stream id
|
||||||
new_stream_id = client.stream.create(name="a shiny new stream")
|
new_stream_id = client.stream.create(name="a shiny new stream")
|
||||||
@@ -52,6 +57,7 @@ class SpeckleClient:
|
|||||||
USE_SSL = True
|
USE_SSL = True
|
||||||
|
|
||||||
def __init__(self, host: str = DEFAULT_HOST, use_ssl: bool = USE_SSL) -> None:
|
def __init__(self, host: str = DEFAULT_HOST, use_ssl: bool = USE_SSL) -> None:
|
||||||
|
metrics.track(metrics.CLIENT, custom_props={"name": "create"})
|
||||||
ws_protocol = "ws"
|
ws_protocol = "ws"
|
||||||
http_protocol = "http"
|
http_protocol = "http"
|
||||||
|
|
||||||
@@ -63,9 +69,9 @@ class SpeckleClient:
|
|||||||
host = re.sub(r"((^\w+:|^)\/\/)|(\/$)", "", host)
|
host = re.sub(r"((^\w+:|^)\/\/)|(\/$)", "", host)
|
||||||
|
|
||||||
self.url = f"{http_protocol}://{host}"
|
self.url = f"{http_protocol}://{host}"
|
||||||
self.graphql = self.url + "/graphql"
|
self.graphql = f"{self.url}/graphql"
|
||||||
self.ws_url = f"{ws_protocol}://{host}/graphql"
|
self.ws_url = f"{ws_protocol}://{host}/graphql"
|
||||||
self.me = None
|
self.account = Account()
|
||||||
|
|
||||||
self.httpclient = Client(
|
self.httpclient = Client(
|
||||||
transport=RequestsHTTPTransport(url=self.graphql, verify=True, retries=3)
|
transport=RequestsHTTPTransport(url=self.graphql, verify=True, retries=3)
|
||||||
@@ -74,19 +80,25 @@ class SpeckleClient:
|
|||||||
|
|
||||||
self._init_resources()
|
self._init_resources()
|
||||||
|
|
||||||
# Check compatibility with the server
|
# ? Check compatibility with the server - i think we can skip this at this point? save a request
|
||||||
try:
|
# try:
|
||||||
serverInfo = self.server.get()
|
# server_info = self.server.get()
|
||||||
if not isinstance(serverInfo, ServerInfo):
|
# if isinstance(server_info, Exception):
|
||||||
raise Exception("Couldn't get ServerInfo")
|
# raise server_info
|
||||||
except Exception as ex:
|
# if not isinstance(server_info, ServerInfo):
|
||||||
raise SpeckleException(f"{self.url} is not a compatible Speckle Server", ex)
|
# raise Exception("Couldn't get ServerInfo")
|
||||||
|
# except Exception as ex:
|
||||||
|
# raise SpeckleException(
|
||||||
|
# f"{self.url} is not a compatible Speckle Server", ex
|
||||||
|
# ) from ex
|
||||||
|
|
||||||
def __repr__(self):
|
def __repr__(self):
|
||||||
return (
|
return f"SpeckleClient( server: {self.url}, authenticated: {self.account.token is not None} )"
|
||||||
f"SpeckleClient( server: {self.url}, authenticated: {self.me is not None} )"
|
|
||||||
)
|
|
||||||
|
|
||||||
|
@deprecated(
|
||||||
|
version="2.6.0",
|
||||||
|
reason="Renamed: please use `authenticate_with_account` or `authenticate_with_token` instead.",
|
||||||
|
)
|
||||||
def authenticate(self, token: str) -> None:
|
def authenticate(self, token: str) -> None:
|
||||||
"""Authenticate the client using a personal access token
|
"""Authenticate the client using a personal access token
|
||||||
The token is saved in the client object and a synchronous GraphQL entrypoint is created
|
The token is saved in the client object and a synchronous GraphQL entrypoint is created
|
||||||
@@ -94,9 +106,35 @@ class SpeckleClient:
|
|||||||
Arguments:
|
Arguments:
|
||||||
token {str} -- an api token
|
token {str} -- an api token
|
||||||
"""
|
"""
|
||||||
self.me = {"token": token}
|
self.authenticate_with_token(token)
|
||||||
|
self._set_up_client()
|
||||||
|
|
||||||
|
def authenticate_with_token(self, token: str) -> None:
|
||||||
|
"""Authenticate the client using a personal access token
|
||||||
|
The token is saved in the client object and a synchronous GraphQL entrypoint is created
|
||||||
|
|
||||||
|
Arguments:
|
||||||
|
token {str} -- an api token
|
||||||
|
"""
|
||||||
|
self.account = get_account_from_token(token, self.url)
|
||||||
|
metrics.track(metrics.CLIENT, self.account, {"name": "authenticate with token"})
|
||||||
|
self._set_up_client()
|
||||||
|
|
||||||
|
def authenticate_with_account(self, account: Account) -> None:
|
||||||
|
"""Authenticate the client using an Account object
|
||||||
|
The account is saved in the client object and a synchronous GraphQL entrypoint is created
|
||||||
|
|
||||||
|
Arguments:
|
||||||
|
account {Account} -- the account object which can be found with `get_default_account` or `get_local_accounts`
|
||||||
|
"""
|
||||||
|
metrics.track(metrics.CLIENT, account, {"name": "authenticate with account"})
|
||||||
|
self.account = account
|
||||||
|
self._set_up_client()
|
||||||
|
|
||||||
|
def _set_up_client(self) -> None:
|
||||||
|
metrics.track(metrics.CLIENT, self.account, {"name": "set up client"})
|
||||||
headers = {
|
headers = {
|
||||||
"Authorization": f"Bearer {self.me['token']}",
|
"Authorization": f"Bearer {self.account.token}",
|
||||||
"Content-Type": "application/json",
|
"Content-Type": "application/json",
|
||||||
}
|
}
|
||||||
httptransport = RequestsHTTPTransport(
|
httptransport = RequestsHTTPTransport(
|
||||||
@@ -104,35 +142,55 @@ class SpeckleClient:
|
|||||||
)
|
)
|
||||||
wstransport = WebsocketsTransport(
|
wstransport = WebsocketsTransport(
|
||||||
url=self.ws_url,
|
url=self.ws_url,
|
||||||
init_payload={"Authorization": f"Bearer {self.me['token']}"},
|
init_payload={"Authorization": f"Bearer {self.account.token}"},
|
||||||
)
|
)
|
||||||
self.httpclient = Client(transport=httptransport)
|
self.httpclient = Client(transport=httptransport)
|
||||||
self.wsclient = Client(transport=wstransport)
|
self.wsclient = Client(transport=wstransport)
|
||||||
|
|
||||||
self._init_resources()
|
self._init_resources()
|
||||||
|
|
||||||
|
if self.user.get() is None:
|
||||||
|
warn(
|
||||||
|
SpeckleWarning(
|
||||||
|
f"Possibly invalid token - could not authenticate Speckle Client for server {self.url}"
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
def execute_query(self, query: str) -> Dict:
|
def execute_query(self, query: str) -> Dict:
|
||||||
return self.httpclient.execute(query)
|
return self.httpclient.execute(query)
|
||||||
|
|
||||||
def _init_resources(self) -> None:
|
def _init_resources(self) -> None:
|
||||||
|
self.server = server.Resource(
|
||||||
|
account=self.account, basepath=self.url, client=self.httpclient
|
||||||
|
)
|
||||||
|
server_version = None
|
||||||
|
try:
|
||||||
|
server_version = self.server.version()
|
||||||
|
except:
|
||||||
|
pass
|
||||||
|
self.user = user.Resource(
|
||||||
|
account=self.account,
|
||||||
|
basepath=self.url,
|
||||||
|
client=self.httpclient,
|
||||||
|
server_version=server_version,
|
||||||
|
)
|
||||||
self.stream = stream.Resource(
|
self.stream = stream.Resource(
|
||||||
me=self.me, basepath=self.url, client=self.httpclient
|
account=self.account,
|
||||||
|
basepath=self.url,
|
||||||
|
client=self.httpclient,
|
||||||
|
server_version=server_version,
|
||||||
)
|
)
|
||||||
self.commit = commit.Resource(
|
self.commit = commit.Resource(
|
||||||
me=self.me, basepath=self.url, client=self.httpclient
|
account=self.account, basepath=self.url, client=self.httpclient
|
||||||
)
|
)
|
||||||
self.branch = branch.Resource(
|
self.branch = branch.Resource(
|
||||||
me=self.me, basepath=self.url, client=self.httpclient
|
account=self.account, basepath=self.url, client=self.httpclient
|
||||||
)
|
)
|
||||||
self.object = object.Resource(
|
self.object = object.Resource(
|
||||||
me=self.me, basepath=self.url, client=self.httpclient
|
account=self.account, basepath=self.url, client=self.httpclient
|
||||||
)
|
)
|
||||||
self.server = server.Resource(
|
|
||||||
me=self.me, basepath=self.url, client=self.httpclient
|
|
||||||
)
|
|
||||||
self.user = user.Resource(me=self.me, basepath=self.url, client=self.httpclient)
|
|
||||||
self.subscribe = subscriptions.Resource(
|
self.subscribe = subscriptions.Resource(
|
||||||
me=self.me,
|
account=self.account,
|
||||||
basepath=self.ws_url,
|
basepath=self.ws_url,
|
||||||
client=self.wsclient,
|
client=self.wsclient,
|
||||||
)
|
)
|
||||||
@@ -140,7 +198,9 @@ class SpeckleClient:
|
|||||||
def __getattr__(self, name):
|
def __getattr__(self, name):
|
||||||
try:
|
try:
|
||||||
attr = getattr(resources, name)
|
attr = getattr(resources, name)
|
||||||
return attr.Resource(me=self.me, basepath=self.url, client=self.httpclient)
|
return attr.Resource(
|
||||||
|
account=self.account, basepath=self.url, client=self.httpclient
|
||||||
|
)
|
||||||
except:
|
except:
|
||||||
raise SpeckleException(
|
raise SpeckleException(
|
||||||
f"Method {name} is not supported by the SpeckleClient class"
|
f"Method {name} is not supported by the SpeckleClient class"
|
||||||
|
|||||||
+58
-105
@@ -1,29 +1,26 @@
|
|||||||
import os
|
import os
|
||||||
from specklepy.transports.server.server import ServerTransport
|
from pydantic import BaseModel, Field # pylint: disable=no-name-in-module
|
||||||
from warnings import warn
|
|
||||||
from pydantic import BaseModel
|
|
||||||
from typing import List, Optional
|
from typing import List, Optional
|
||||||
from urllib.parse import urlparse, unquote
|
from specklepy.logging import metrics
|
||||||
from specklepy.api.models import ServerInfo
|
from specklepy.api.models import ServerInfo
|
||||||
from specklepy.api.client import SpeckleClient
|
|
||||||
from specklepy.transports.sqlite import SQLiteTransport
|
from specklepy.transports.sqlite import SQLiteTransport
|
||||||
from specklepy.logging.exceptions import SpeckleException, SpeckleWarning
|
from specklepy.logging.exceptions import SpeckleException
|
||||||
|
|
||||||
|
|
||||||
class UserInfo(BaseModel):
|
class UserInfo(BaseModel):
|
||||||
name: str
|
name: Optional[str]
|
||||||
email: str
|
email: Optional[str]
|
||||||
company: Optional[str]
|
company: Optional[str]
|
||||||
id: str
|
id: Optional[str]
|
||||||
|
|
||||||
|
|
||||||
class Account(BaseModel):
|
class Account(BaseModel):
|
||||||
isDefault: bool = None
|
isDefault: bool = False
|
||||||
token: str
|
token: Optional[str] = None
|
||||||
refreshToken: str = None
|
refreshToken: Optional[str] = None
|
||||||
serverInfo: ServerInfo
|
serverInfo: ServerInfo = Field(default_factory=ServerInfo)
|
||||||
userInfo: UserInfo
|
userInfo: UserInfo = Field(default_factory=UserInfo)
|
||||||
id: str = None
|
id: Optional[str] = None
|
||||||
|
|
||||||
def __repr__(self) -> str:
|
def __repr__(self) -> str:
|
||||||
return f"Account(email: {self.userInfo.email}, server: {self.serverInfo.url}, isDefault: {self.isDefault})"
|
return f"Account(email: {self.userInfo.email}, server: {self.serverInfo.url}, isDefault: {self.isDefault})"
|
||||||
@@ -31,6 +28,12 @@ class Account(BaseModel):
|
|||||||
def __str__(self) -> str:
|
def __str__(self) -> str:
|
||||||
return self.__repr__()
|
return self.__repr__()
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def from_token(cls, token: str, server_url: str = None):
|
||||||
|
acct = cls(token=token)
|
||||||
|
acct.serverInfo.url = server_url
|
||||||
|
return acct
|
||||||
|
|
||||||
|
|
||||||
def get_local_accounts(base_path: str = None) -> List[Account]:
|
def get_local_accounts(base_path: str = None) -> List[Account]:
|
||||||
"""Gets all the accounts present in this environment
|
"""Gets all the accounts present in this environment
|
||||||
@@ -42,12 +45,15 @@ def get_local_accounts(base_path: str = None) -> List[Account]:
|
|||||||
List[Account] -- list of all local accounts or an empty list if no accounts were found
|
List[Account] -- list of all local accounts or an empty list if no accounts were found
|
||||||
"""
|
"""
|
||||||
account_storage = SQLiteTransport(scope="Accounts", base_path=base_path)
|
account_storage = SQLiteTransport(scope="Accounts", base_path=base_path)
|
||||||
|
# pylint: disable=protected-access
|
||||||
json_path = os.path.join(account_storage._base_path, "Accounts")
|
json_path = os.path.join(account_storage._base_path, "Accounts")
|
||||||
os.makedirs(json_path, exist_ok=True)
|
os.makedirs(json_path, exist_ok=True)
|
||||||
json_acct_files = [file for file in os.listdir(json_path) if file.endswith(".json")]
|
json_acct_files = [file for file in os.listdir(json_path) if file.endswith(".json")]
|
||||||
|
|
||||||
accounts = []
|
accounts: List[Account] = []
|
||||||
res = account_storage.get_all_objects()
|
res = account_storage.get_all_objects()
|
||||||
|
account_storage.close()
|
||||||
|
|
||||||
if res:
|
if res:
|
||||||
accounts.extend(Account.parse_raw(r[1]) for r in res)
|
accounts.extend(Account.parse_raw(r[1]) for r in res)
|
||||||
if json_acct_files:
|
if json_acct_files:
|
||||||
@@ -60,7 +66,15 @@ def get_local_accounts(base_path: str = None) -> List[Account]:
|
|||||||
raise SpeckleException(
|
raise SpeckleException(
|
||||||
"Invalid json accounts could not be read. Please fix or remove them.",
|
"Invalid json accounts could not be read. Please fix or remove them.",
|
||||||
ex,
|
ex,
|
||||||
)
|
) from ex
|
||||||
|
|
||||||
|
metrics.track(
|
||||||
|
metrics.ACCOUNTS,
|
||||||
|
next(
|
||||||
|
(acc for acc in accounts if acc.isDefault),
|
||||||
|
accounts[0] if accounts else None,
|
||||||
|
),
|
||||||
|
)
|
||||||
|
|
||||||
return accounts
|
return accounts
|
||||||
|
|
||||||
@@ -81,102 +95,41 @@ def get_default_account(base_path: str = None) -> Account:
|
|||||||
if not default:
|
if not default:
|
||||||
default = accounts[0]
|
default = accounts[0]
|
||||||
default.isDefault = True
|
default.isDefault = True
|
||||||
|
metrics.initialise_tracker(default)
|
||||||
|
|
||||||
return default
|
return default
|
||||||
|
|
||||||
|
|
||||||
class StreamWrapper:
|
def get_account_from_token(token: str, server_url: str = None) -> Account:
|
||||||
stream_url: str = None
|
"""Gets the local account for the token if it exists
|
||||||
use_ssl: bool = True
|
Arguments:
|
||||||
host: str = None
|
token {str} -- the api token
|
||||||
stream_id: str = None
|
|
||||||
commit_id: str = None
|
|
||||||
object_id: str = None
|
|
||||||
branch_name: str = None
|
|
||||||
client: SpeckleClient = None
|
|
||||||
account: Account = None
|
|
||||||
|
|
||||||
def __repr__(self):
|
Returns:
|
||||||
return f"StreamWrapper( server: {self.host}, stream_id: {self.stream_id}, type: {self.type} )"
|
Account -- the local account with this token or a shell account containing just the token and url if no local account is found
|
||||||
|
"""
|
||||||
|
accounts = get_local_accounts()
|
||||||
|
if not accounts:
|
||||||
|
return Account.from_token(token, server_url)
|
||||||
|
|
||||||
def __str__(self) -> str:
|
acct = next((acc for acc in accounts if acc.token == token), None)
|
||||||
return self.__repr__()
|
if acct:
|
||||||
|
return acct
|
||||||
|
|
||||||
@property
|
if server_url:
|
||||||
def type(self) -> str:
|
url = server_url.lower()
|
||||||
if self.object_id:
|
acct = next(
|
||||||
return "object"
|
(acc for acc in accounts if url in acc.serverInfo.url.lower()), None
|
||||||
elif self.commit_id:
|
|
||||||
return "commit"
|
|
||||||
elif self.branch_name:
|
|
||||||
return "branch"
|
|
||||||
else:
|
|
||||||
return "stream" if self.stream_id else "invalid"
|
|
||||||
|
|
||||||
def __init__(self, url: str) -> None:
|
|
||||||
self.stream_url = url
|
|
||||||
parsed = urlparse(url)
|
|
||||||
self.host = parsed.netloc
|
|
||||||
self.use_ssl = parsed.scheme == "https"
|
|
||||||
segments = parsed.path.strip("/").split("/")
|
|
||||||
|
|
||||||
if not segments or len(segments) > 4 or len(segments) < 2:
|
|
||||||
raise SpeckleException(
|
|
||||||
f"Cannot parse {url} into a stream wrapper class - invalid URL provided."
|
|
||||||
)
|
|
||||||
|
|
||||||
while segments:
|
|
||||||
segment = segments.pop(0)
|
|
||||||
if segments and segment.lower() == "streams":
|
|
||||||
self.stream_id = segments.pop(0)
|
|
||||||
elif segments and segment.lower() == "commits":
|
|
||||||
self.commit_id = segments.pop(0)
|
|
||||||
elif segments and segment.lower() == "branches":
|
|
||||||
self.branch_name = unquote(segments.pop(0))
|
|
||||||
elif segments and segment.lower() == "objects":
|
|
||||||
self.object_id = segments.pop(0)
|
|
||||||
elif segment.lower() == "globals":
|
|
||||||
self.branch_name = "globals"
|
|
||||||
if segments:
|
|
||||||
self.commit_id = segments.pop(0)
|
|
||||||
else:
|
|
||||||
raise SpeckleException(
|
|
||||||
f"Cannot parse {url} into a stream wrapper class - invalid URL provided."
|
|
||||||
)
|
|
||||||
|
|
||||||
if not self.stream_id:
|
|
||||||
raise SpeckleException(
|
|
||||||
f"Cannot parse {url} into a stream wrapper class - no stream id found."
|
|
||||||
)
|
|
||||||
|
|
||||||
def get_account(self) -> Account:
|
|
||||||
if self.account:
|
|
||||||
return self.account
|
|
||||||
|
|
||||||
self.account = next(
|
|
||||||
(a for a in get_local_accounts() if self.host in a.serverInfo.url),
|
|
||||||
None,
|
|
||||||
)
|
)
|
||||||
|
if acct:
|
||||||
|
return acct
|
||||||
|
|
||||||
return self.account
|
return Account.from_token(token, server_url)
|
||||||
|
|
||||||
def get_client(self) -> SpeckleClient:
|
|
||||||
if self.client:
|
|
||||||
return self.client
|
|
||||||
|
|
||||||
if not self.account:
|
class StreamWrapper:
|
||||||
self.get_account()
|
def __init__(self, url: str = None) -> None:
|
||||||
|
raise SpeckleException(
|
||||||
self.client = SpeckleClient(host=self.host, use_ssl=self.use_ssl)
|
message="The StreamWrapper has moved as of v2.6.0! Please import from specklepy.api.wrapper",
|
||||||
|
exception=DeprecationWarning,
|
||||||
if self.account is None:
|
)
|
||||||
warn(f"No local account found for server {self.host}", SpeckleWarning)
|
|
||||||
return self.client
|
|
||||||
|
|
||||||
self.client.authenticate(self.account.token)
|
|
||||||
return self.client
|
|
||||||
|
|
||||||
def get_transport(self) -> ServerTransport:
|
|
||||||
if not self.client:
|
|
||||||
self.get_client()
|
|
||||||
return ServerTransport(self.client, self.stream_id)
|
|
||||||
|
|||||||
+60
-9
@@ -3,10 +3,10 @@
|
|||||||
# timestamp: 2020-11-17T14:33:13+00:00
|
# timestamp: 2020-11-17T14:33:13+00:00
|
||||||
|
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
from typing import Any, Dict, List, Optional
|
from typing import List, Optional
|
||||||
|
|
||||||
|
|
||||||
from pydantic import BaseModel
|
from pydantic import BaseModel # pylint: disable=no-name-in-module
|
||||||
|
|
||||||
|
|
||||||
class Collaborator(BaseModel):
|
class Collaborator(BaseModel):
|
||||||
@@ -23,14 +23,14 @@ class Commit(BaseModel):
|
|||||||
authorId: Optional[str]
|
authorId: Optional[str]
|
||||||
authorAvatar: Optional[str]
|
authorAvatar: Optional[str]
|
||||||
branchName: Optional[str]
|
branchName: Optional[str]
|
||||||
createdAt: Optional[str]
|
createdAt: Optional[datetime]
|
||||||
sourceApplication: Optional[str]
|
sourceApplication: Optional[str]
|
||||||
referencedObject: Optional[str]
|
referencedObject: Optional[str]
|
||||||
totalChildrenCount: Optional[int]
|
totalChildrenCount: Optional[int]
|
||||||
parents: Optional[List[str]]
|
parents: Optional[List[str]]
|
||||||
|
|
||||||
def __repr__(self) -> str:
|
def __repr__(self) -> str:
|
||||||
return f"Commit( id: {self.id}, message: {self.message}, referencedObject: {self.referencedObject}, authorName: {self.authorName}, createdAt: {self.createdAt} )"
|
return f"Commit( id: {self.id}, message: {self.message}, referencedObject: {self.referencedObject}, authorName: {self.authorName}, branchName: {self.branchName}, createdAt: {self.createdAt} )"
|
||||||
|
|
||||||
def __str__(self) -> str:
|
def __str__(self) -> str:
|
||||||
return self.__repr__()
|
return self.__repr__()
|
||||||
@@ -38,7 +38,7 @@ class Commit(BaseModel):
|
|||||||
|
|
||||||
class Commits(BaseModel):
|
class Commits(BaseModel):
|
||||||
totalCount: Optional[int]
|
totalCount: Optional[int]
|
||||||
cursor: Optional[Any]
|
cursor: Optional[datetime]
|
||||||
items: List[Commit] = []
|
items: List[Commit] = []
|
||||||
|
|
||||||
|
|
||||||
@@ -47,7 +47,7 @@ class Object(BaseModel):
|
|||||||
speckleType: Optional[str]
|
speckleType: Optional[str]
|
||||||
applicationId: Optional[str]
|
applicationId: Optional[str]
|
||||||
totalChildrenCount: Optional[int]
|
totalChildrenCount: Optional[int]
|
||||||
createdAt: Optional[str]
|
createdAt: Optional[datetime]
|
||||||
|
|
||||||
|
|
||||||
class Branch(BaseModel):
|
class Branch(BaseModel):
|
||||||
@@ -66,14 +66,18 @@ class Branches(BaseModel):
|
|||||||
class Stream(BaseModel):
|
class Stream(BaseModel):
|
||||||
id: Optional[str]
|
id: Optional[str]
|
||||||
name: Optional[str]
|
name: Optional[str]
|
||||||
description: Optional[str]
|
role: Optional[str]
|
||||||
isPublic: Optional[bool]
|
isPublic: Optional[bool]
|
||||||
createdAt: Optional[str]
|
description: Optional[str]
|
||||||
updatedAt: Optional[str]
|
createdAt: Optional[datetime]
|
||||||
|
updatedAt: Optional[datetime]
|
||||||
collaborators: List[Collaborator] = []
|
collaborators: List[Collaborator] = []
|
||||||
branches: Optional[Branches]
|
branches: Optional[Branches]
|
||||||
commit: Optional[Commit]
|
commit: Optional[Commit]
|
||||||
object: Optional[Object]
|
object: Optional[Object]
|
||||||
|
commentCount: Optional[int]
|
||||||
|
favoritedDate: Optional[datetime]
|
||||||
|
favoritesCount: Optional[int]
|
||||||
|
|
||||||
def __repr__(self):
|
def __repr__(self):
|
||||||
return f"Stream( id: {self.id}, name: {self.name}, description: {self.description}, isPublic: {self.isPublic})"
|
return f"Stream( id: {self.id}, name: {self.name}, description: {self.description}, isPublic: {self.isPublic})"
|
||||||
@@ -106,6 +110,53 @@ class User(BaseModel):
|
|||||||
return self.__repr__()
|
return self.__repr__()
|
||||||
|
|
||||||
|
|
||||||
|
class PendingStreamCollaborator(BaseModel):
|
||||||
|
id: Optional[str]
|
||||||
|
inviteId: Optional[str]
|
||||||
|
streamId: Optional[str]
|
||||||
|
streamName: Optional[str]
|
||||||
|
title: Optional[str]
|
||||||
|
role: Optional[str]
|
||||||
|
invitedBy: Optional[User]
|
||||||
|
user: Optional[User]
|
||||||
|
token: Optional[str]
|
||||||
|
|
||||||
|
def __repr__(self):
|
||||||
|
return f"PendingStreamCollaborator( inviteId: {self.inviteId}, streamId: {self.streamId}, role: {self.role}, title: {self.title}, invitedBy: {self.user.name if self.user else None})"
|
||||||
|
|
||||||
|
def __str__(self) -> str:
|
||||||
|
return self.__repr__()
|
||||||
|
|
||||||
|
|
||||||
|
class Activity(BaseModel):
|
||||||
|
actionType: Optional[str]
|
||||||
|
info: Optional[dict]
|
||||||
|
userId: Optional[str]
|
||||||
|
streamId: Optional[str]
|
||||||
|
resourceId: Optional[str]
|
||||||
|
resourceType: Optional[str]
|
||||||
|
message: Optional[str]
|
||||||
|
time: Optional[datetime]
|
||||||
|
|
||||||
|
def __repr__(self) -> str:
|
||||||
|
return f"Activity( streamId: {self.streamId}, actionType: {self.actionType}, message: {self.message}, userId: {self.userId} )"
|
||||||
|
|
||||||
|
def __str__(self) -> str:
|
||||||
|
return self.__repr__()
|
||||||
|
|
||||||
|
|
||||||
|
class ActivityCollection(BaseModel):
|
||||||
|
totalCount: Optional[int]
|
||||||
|
items: Optional[List[Activity]]
|
||||||
|
cursor: Optional[datetime]
|
||||||
|
|
||||||
|
def __repr__(self) -> str:
|
||||||
|
return f"ActivityCollection( totalCount: {self.totalCount}, items: {len(self.items) if self.items else 0}, cursor: {self.cursor.isoformat() if self.cursor else None} )"
|
||||||
|
|
||||||
|
def __str__(self) -> str:
|
||||||
|
return self.__repr__()
|
||||||
|
|
||||||
|
|
||||||
class ServerInfo(BaseModel):
|
class ServerInfo(BaseModel):
|
||||||
name: Optional[str]
|
name: Optional[str]
|
||||||
company: Optional[str]
|
company: Optional[str]
|
||||||
|
|||||||
+19
-10
@@ -1,4 +1,5 @@
|
|||||||
from typing import List
|
from typing import List
|
||||||
|
from specklepy.logging import metrics
|
||||||
from specklepy.objects.base import Base
|
from specklepy.objects.base import Base
|
||||||
from specklepy.transports.sqlite import SQLiteTransport
|
from specklepy.transports.sqlite import SQLiteTransport
|
||||||
from specklepy.transports.server import ServerTransport
|
from specklepy.transports.server import ServerTransport
|
||||||
@@ -9,7 +10,7 @@ from specklepy.serialization.base_object_serializer import BaseObjectSerializer
|
|||||||
|
|
||||||
def send(
|
def send(
|
||||||
base: Base,
|
base: Base,
|
||||||
transports: List[AbstractTransport] = [],
|
transports: List[AbstractTransport] = None,
|
||||||
use_default_cache: bool = True,
|
use_default_cache: bool = True,
|
||||||
):
|
):
|
||||||
"""Sends an object via the provided transports. Defaults to the local cache.
|
"""Sends an object via the provided transports. Defaults to the local cache.
|
||||||
@@ -22,23 +23,29 @@ def send(
|
|||||||
Returns:
|
Returns:
|
||||||
str -- the object id of the sent object
|
str -- the object id of the sent object
|
||||||
"""
|
"""
|
||||||
|
|
||||||
if not transports and not use_default_cache:
|
if not transports and not use_default_cache:
|
||||||
raise SpeckleException(
|
raise SpeckleException(
|
||||||
message="You need to provide at least one transport: cannot send with an empty transport list and no default cache"
|
message="You need to provide at least one transport: cannot send with an empty transport list and no default cache"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
if isinstance(transports, AbstractTransport):
|
||||||
|
transports = [transports]
|
||||||
|
|
||||||
|
if transports is None:
|
||||||
|
metrics.track(metrics.SEND)
|
||||||
|
transports = []
|
||||||
|
else:
|
||||||
|
metrics.track(metrics.SEND, getattr(transports[0], "account", None))
|
||||||
|
|
||||||
if use_default_cache:
|
if use_default_cache:
|
||||||
transports.insert(0, SQLiteTransport())
|
transports.insert(0, SQLiteTransport())
|
||||||
|
|
||||||
serializer = BaseObjectSerializer(write_transports=transports)
|
serializer = BaseObjectSerializer(write_transports=transports)
|
||||||
|
|
||||||
for t in transports:
|
obj_hash, _ = serializer.write_json(base=base)
|
||||||
t.begin_write()
|
|
||||||
hash, _ = serializer.write_json(base=base)
|
|
||||||
|
|
||||||
for t in transports:
|
return obj_hash
|
||||||
t.end_write()
|
|
||||||
|
|
||||||
return hash
|
|
||||||
|
|
||||||
|
|
||||||
def receive(
|
def receive(
|
||||||
@@ -57,13 +64,13 @@ def receive(
|
|||||||
Returns:
|
Returns:
|
||||||
Base -- the base object
|
Base -- the base object
|
||||||
"""
|
"""
|
||||||
|
metrics.track(metrics.RECEIVE, getattr(remote_transport, "account", None))
|
||||||
if not local_transport:
|
if not local_transport:
|
||||||
local_transport = SQLiteTransport()
|
local_transport = SQLiteTransport()
|
||||||
|
|
||||||
serializer = BaseObjectSerializer(read_transport=local_transport)
|
serializer = BaseObjectSerializer(read_transport=local_transport)
|
||||||
|
|
||||||
# try local transport first. if the parent is there, we assume all the children are there and continue wth deserialisation using the local transport
|
# try local transport first. if the parent is there, we assume all the children are there and continue with deserialisation using the local transport
|
||||||
obj_string = local_transport.get_object(obj_id)
|
obj_string = local_transport.get_object(obj_id)
|
||||||
if obj_string:
|
if obj_string:
|
||||||
return serializer.read_json(obj_string=obj_string)
|
return serializer.read_json(obj_string=obj_string)
|
||||||
@@ -92,6 +99,7 @@ def serialize(base: Base, write_transports: List[AbstractTransport] = []) -> str
|
|||||||
Returns:
|
Returns:
|
||||||
str -- the serialized object
|
str -- the serialized object
|
||||||
"""
|
"""
|
||||||
|
metrics.track(metrics.SERIALIZE)
|
||||||
serializer = BaseObjectSerializer(write_transports=write_transports)
|
serializer = BaseObjectSerializer(write_transports=write_transports)
|
||||||
|
|
||||||
return serializer.write_json(base)[1]
|
return serializer.write_json(base)[1]
|
||||||
@@ -109,6 +117,7 @@ def deserialize(obj_string: str, read_transport: AbstractTransport = None) -> Ba
|
|||||||
Returns:
|
Returns:
|
||||||
Base -- the deserialized object
|
Base -- the deserialized object
|
||||||
"""
|
"""
|
||||||
|
metrics.track(metrics.DESERIALIZE)
|
||||||
if not read_transport:
|
if not read_transport:
|
||||||
read_transport = SQLiteTransport()
|
read_transport = SQLiteTransport()
|
||||||
|
|
||||||
|
|||||||
+55
-22
@@ -1,41 +1,48 @@
|
|||||||
|
from graphql import DocumentNode
|
||||||
|
from specklepy.api.credentials import Account
|
||||||
from specklepy.transports.sqlite import SQLiteTransport
|
from specklepy.transports.sqlite import SQLiteTransport
|
||||||
from typing import Dict, List
|
from typing import Any, Dict, List, Optional, Tuple, Type, Union
|
||||||
from gql.client import Client
|
from gql.client import Client
|
||||||
from gql.gql import gql
|
|
||||||
from gql.transport.exceptions import TransportQueryError
|
from gql.transport.exceptions import TransportQueryError
|
||||||
from specklepy.logging.exceptions import GraphQLException, SpeckleException
|
from specklepy.logging.exceptions import (
|
||||||
|
GraphQLException,
|
||||||
|
SpeckleException,
|
||||||
|
UnsupportedException,
|
||||||
|
)
|
||||||
from specklepy.serialization.base_object_serializer import BaseObjectSerializer
|
from specklepy.serialization.base_object_serializer import BaseObjectSerializer
|
||||||
|
|
||||||
|
|
||||||
class ResourceBase(object):
|
class ResourceBase(object):
|
||||||
def __init__(
|
def __init__(
|
||||||
self,
|
self,
|
||||||
me: Dict,
|
account: Account,
|
||||||
basepath: str,
|
basepath: str,
|
||||||
client: Client,
|
client: Client,
|
||||||
name: str,
|
name: str,
|
||||||
methods: list,
|
server_version: Optional[Tuple[Any, ...]] = None,
|
||||||
) -> None:
|
) -> None:
|
||||||
self.me = me
|
self.account = account
|
||||||
self.basepath = basepath
|
self.basepath = basepath
|
||||||
self.client = client
|
self.client = client
|
||||||
self.name = name
|
self.name = name
|
||||||
self.methods = methods
|
self.server_version = server_version
|
||||||
self.schema = None
|
self.schema: Optional[Type] = None
|
||||||
|
|
||||||
def _step_into_response(self, response: dict, return_type: str or List):
|
def _step_into_response(self, response: dict, return_type: Union[str, List, None]):
|
||||||
"""Step into the dict to get the relevant data"""
|
"""Step into the dict to get the relevant data"""
|
||||||
if return_type is None:
|
if return_type is None:
|
||||||
return response
|
return response
|
||||||
elif isinstance(return_type, str):
|
if isinstance(return_type, str):
|
||||||
return response[return_type]
|
return response[return_type]
|
||||||
elif isinstance(return_type, List):
|
if isinstance(return_type, List):
|
||||||
for key in return_type:
|
for key in return_type:
|
||||||
response = response[key]
|
response = response[key]
|
||||||
return response
|
return response
|
||||||
|
|
||||||
def _parse_response(self, response: dict or list, schema=None):
|
def _parse_response(self, response: Union[dict, list, None], schema=None):
|
||||||
"""Try to create a class instance from the response"""
|
"""Try to create a class instance from the response"""
|
||||||
|
if response is None:
|
||||||
|
return None
|
||||||
if isinstance(response, list):
|
if isinstance(response, list):
|
||||||
return [self._parse_response(response=r, schema=schema) for r in response]
|
return [self._parse_response(response=r, schema=schema) for r in response]
|
||||||
if schema:
|
if schema:
|
||||||
@@ -51,26 +58,26 @@ class ResourceBase(object):
|
|||||||
|
|
||||||
def make_request(
|
def make_request(
|
||||||
self,
|
self,
|
||||||
query: gql,
|
query: DocumentNode,
|
||||||
params: Dict = None,
|
params: Dict = None,
|
||||||
return_type: str or List = None,
|
return_type: Union[str, List, None] = None,
|
||||||
schema=None,
|
schema=None,
|
||||||
parse_response: bool = True,
|
parse_response: bool = True,
|
||||||
) -> Dict or GraphQLException:
|
) -> Any:
|
||||||
"""Executes the GraphQL query"""
|
"""Executes the GraphQL query"""
|
||||||
try:
|
try:
|
||||||
response = self.client.execute(query, variable_values=params)
|
response = self.client.execute(query, variable_values=params)
|
||||||
except Exception as e:
|
except Exception as ex:
|
||||||
if isinstance(e, TransportQueryError):
|
if isinstance(ex, TransportQueryError):
|
||||||
return GraphQLException(
|
return GraphQLException(
|
||||||
message=f"Failed to execute the GraphQL {self.name} request. Errors: {e.errors}",
|
message=f"Failed to execute the GraphQL {self.name} request. Errors: {ex.errors}",
|
||||||
errors=e.errors,
|
errors=ex.errors,
|
||||||
data=e.data,
|
data=ex.data,
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
return SpeckleException(
|
return SpeckleException(
|
||||||
message=f"Failed to execute the GraphQL {self.name} request. Inner exception: {e}",
|
message=f"Failed to execute the GraphQL {self.name} request. Inner exception: {ex}",
|
||||||
exception=e,
|
exception=ex,
|
||||||
)
|
)
|
||||||
|
|
||||||
response = self._step_into_response(response=response, return_type=return_type)
|
response = self._step_into_response(response=response, return_type=return_type)
|
||||||
@@ -79,3 +86,29 @@ class ResourceBase(object):
|
|||||||
return self._parse_response(response=response, schema=schema)
|
return self._parse_response(response=response, schema=schema)
|
||||||
else:
|
else:
|
||||||
return response
|
return response
|
||||||
|
|
||||||
|
def _check_server_version_at_least(
|
||||||
|
self, target_version: Tuple[Any, ...], unsupported_message: str = None
|
||||||
|
):
|
||||||
|
"""Use this check to guard against making unsupported requests on older servers.
|
||||||
|
|
||||||
|
Arguments:
|
||||||
|
target_version {tuple} -- the minimum server version in the format (major, minor, patch, (tag, build))
|
||||||
|
eg (2, 6, 3) for a stable build and (2, 6, 4, 'alpha', 4711) for alpha
|
||||||
|
"""
|
||||||
|
if not unsupported_message:
|
||||||
|
unsupported_message = f"The client method used is not supported on Speckle Server versios prior to v{'.'.join(target_version)}"
|
||||||
|
if self.server_version and self.server_version < target_version:
|
||||||
|
raise UnsupportedException(unsupported_message)
|
||||||
|
|
||||||
|
def _check_invites_supported(self):
|
||||||
|
"""Invites are only supported for Speckle Server >= 2.6.4.
|
||||||
|
Use this check to guard against making unsupported requests on older servers.
|
||||||
|
"""
|
||||||
|
self._check_server_version_at_least(
|
||||||
|
(2, 6, 4),
|
||||||
|
(
|
||||||
|
"Stream invites are only supported as of Speckle Server v2.6.4. "
|
||||||
|
"Please update your Speckle Server to use this method or use the `grant_permission` flow instead."
|
||||||
|
),
|
||||||
|
)
|
||||||
|
|||||||
@@ -1,6 +1,5 @@
|
|||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
import sys
|
import sys
|
||||||
import inspect
|
|
||||||
import pkgutil
|
import pkgutil
|
||||||
from importlib import import_module
|
from importlib import import_module
|
||||||
|
|
||||||
|
|||||||
@@ -1,20 +1,20 @@
|
|||||||
from specklepy.api.resources import stream
|
|
||||||
from typing import List, Optional
|
|
||||||
from gql import gql
|
from gql import gql
|
||||||
from pydantic.main import BaseModel
|
|
||||||
from specklepy.api.resource import ResourceBase
|
from specklepy.api.resource import ResourceBase
|
||||||
from specklepy.api.models import Branch
|
from specklepy.api.models import Branch
|
||||||
|
from specklepy.logging import metrics
|
||||||
|
|
||||||
NAME = "branch"
|
NAME = "branch"
|
||||||
METHODS = ["create"]
|
|
||||||
|
|
||||||
|
|
||||||
class Resource(ResourceBase):
|
class Resource(ResourceBase):
|
||||||
"""API Access class for branches"""
|
"""API Access class for branches"""
|
||||||
|
|
||||||
def __init__(self, me, basepath, client) -> None:
|
def __init__(self, account, basepath, client) -> None:
|
||||||
super().__init__(
|
super().__init__(
|
||||||
me=me, basepath=basepath, client=client, name=NAME, methods=METHODS
|
account=account,
|
||||||
|
basepath=basepath,
|
||||||
|
client=client,
|
||||||
|
name=NAME,
|
||||||
)
|
)
|
||||||
self.schema = Branch
|
self.schema = Branch
|
||||||
|
|
||||||
@@ -30,7 +30,7 @@ class Resource(ResourceBase):
|
|||||||
Returns:
|
Returns:
|
||||||
id {str} -- the newly created branch's id
|
id {str} -- the newly created branch's id
|
||||||
"""
|
"""
|
||||||
|
metrics.track(metrics.BRANCH, self.account, {"name": "create"})
|
||||||
query = gql(
|
query = gql(
|
||||||
"""
|
"""
|
||||||
mutation BranchCreate($branch: BranchCreateInput!) {
|
mutation BranchCreate($branch: BranchCreateInput!) {
|
||||||
@@ -61,7 +61,7 @@ class Resource(ResourceBase):
|
|||||||
Returns:
|
Returns:
|
||||||
Branch -- the fetched branch with its latest commits
|
Branch -- the fetched branch with its latest commits
|
||||||
"""
|
"""
|
||||||
|
metrics.track(metrics.BRANCH, self.account, {"name": "get"})
|
||||||
query = gql(
|
query = gql(
|
||||||
"""
|
"""
|
||||||
query BranchGet($stream_id: String!, $name: String!, $commits_limit: Int!) {
|
query BranchGet($stream_id: String!, $name: String!, $commits_limit: Int!) {
|
||||||
@@ -109,6 +109,7 @@ class Resource(ResourceBase):
|
|||||||
Returns:
|
Returns:
|
||||||
List[Branch] -- the branches on the stream
|
List[Branch] -- the branches on the stream
|
||||||
"""
|
"""
|
||||||
|
metrics.track(metrics.BRANCH, self.account, {"name": "get"})
|
||||||
query = gql(
|
query = gql(
|
||||||
"""
|
"""
|
||||||
query BranchesGet($stream_id: String!, $branches_limit: Int!, $commits_limit: Int!) {
|
query BranchesGet($stream_id: String!, $branches_limit: Int!, $commits_limit: Int!) {
|
||||||
@@ -161,8 +162,9 @@ class Resource(ResourceBase):
|
|||||||
description {str} -- optional: the updated branch description
|
description {str} -- optional: the updated branch description
|
||||||
|
|
||||||
Returns:
|
Returns:
|
||||||
bool -- True if update is successfull
|
bool -- True if update is successful
|
||||||
"""
|
"""
|
||||||
|
metrics.track(metrics.BRANCH, self.account, {"name": "update"})
|
||||||
query = gql(
|
query = gql(
|
||||||
"""
|
"""
|
||||||
mutation BranchUpdate($branch: BranchUpdateInput!) {
|
mutation BranchUpdate($branch: BranchUpdateInput!) {
|
||||||
@@ -196,7 +198,7 @@ class Resource(ResourceBase):
|
|||||||
Returns:
|
Returns:
|
||||||
bool -- True if deletion is successful
|
bool -- True if deletion is successful
|
||||||
"""
|
"""
|
||||||
|
metrics.track(metrics.BRANCH, self.account, {"name": "delete"})
|
||||||
query = gql(
|
query = gql(
|
||||||
"""
|
"""
|
||||||
mutation BranchDelete($branch: BranchDeleteInput!) {
|
mutation BranchDelete($branch: BranchDeleteInput!) {
|
||||||
|
|||||||
@@ -2,18 +2,21 @@ from typing import Optional, List
|
|||||||
from gql import gql
|
from gql import gql
|
||||||
from specklepy.api.resource import ResourceBase
|
from specklepy.api.resource import ResourceBase
|
||||||
from specklepy.api.models import Commit
|
from specklepy.api.models import Commit
|
||||||
|
from specklepy.logging import metrics
|
||||||
|
|
||||||
|
|
||||||
NAME = "commit"
|
NAME = "commit"
|
||||||
METHODS = []
|
|
||||||
|
|
||||||
|
|
||||||
class Resource(ResourceBase):
|
class Resource(ResourceBase):
|
||||||
"""API Access class for commits"""
|
"""API Access class for commits"""
|
||||||
|
|
||||||
def __init__(self, me, basepath, client) -> None:
|
def __init__(self, account, basepath, client) -> None:
|
||||||
super().__init__(
|
super().__init__(
|
||||||
me=me, basepath=basepath, client=client, name=NAME, methods=METHODS
|
account=account,
|
||||||
|
basepath=basepath,
|
||||||
|
client=client,
|
||||||
|
name=NAME,
|
||||||
)
|
)
|
||||||
self.schema = Commit
|
self.schema = Commit
|
||||||
|
|
||||||
@@ -34,8 +37,8 @@ class Resource(ResourceBase):
|
|||||||
stream(id: $stream_id) {
|
stream(id: $stream_id) {
|
||||||
commit(id: $commit_id) {
|
commit(id: $commit_id) {
|
||||||
id
|
id
|
||||||
referencedObject
|
|
||||||
message
|
message
|
||||||
|
referencedObject
|
||||||
authorId
|
authorId
|
||||||
authorName
|
authorName
|
||||||
authorAvatar
|
authorAvatar
|
||||||
@@ -66,6 +69,7 @@ class Resource(ResourceBase):
|
|||||||
Returns:
|
Returns:
|
||||||
List[Commit] -- a list of the most recent commit objects
|
List[Commit] -- a list of the most recent commit objects
|
||||||
"""
|
"""
|
||||||
|
metrics.track(metrics.COMMIT, self.account, {"name": "get"})
|
||||||
query = gql(
|
query = gql(
|
||||||
"""
|
"""
|
||||||
query Commits($stream_id: String!, $limit: Int!) {
|
query Commits($stream_id: String!, $limit: Int!) {
|
||||||
@@ -79,6 +83,7 @@ class Resource(ResourceBase):
|
|||||||
authorId
|
authorId
|
||||||
authorName
|
authorName
|
||||||
authorAvatar
|
authorAvatar
|
||||||
|
branchName
|
||||||
createdAt
|
createdAt
|
||||||
sourceApplication
|
sourceApplication
|
||||||
totalChildrenCount
|
totalChildrenCount
|
||||||
@@ -118,6 +123,7 @@ class Resource(ResourceBase):
|
|||||||
Returns:
|
Returns:
|
||||||
str -- the id of the created commit
|
str -- the id of the created commit
|
||||||
"""
|
"""
|
||||||
|
metrics.track(metrics.COMMIT, self.account, {"name": "create"})
|
||||||
query = gql(
|
query = gql(
|
||||||
"""
|
"""
|
||||||
mutation CommitCreate ($commit: CommitCreateInput!){ commitCreate(commit: $commit)}
|
mutation CommitCreate ($commit: CommitCreateInput!){ commitCreate(commit: $commit)}
|
||||||
@@ -151,6 +157,7 @@ class Resource(ResourceBase):
|
|||||||
Returns:
|
Returns:
|
||||||
bool -- True if the operation succeeded
|
bool -- True if the operation succeeded
|
||||||
"""
|
"""
|
||||||
|
metrics.track(metrics.COMMIT, self.account, {"name": "update"})
|
||||||
query = gql(
|
query = gql(
|
||||||
"""
|
"""
|
||||||
mutation CommitUpdate($commit: CommitUpdateInput!){ commitUpdate(commit: $commit)}
|
mutation CommitUpdate($commit: CommitUpdateInput!){ commitUpdate(commit: $commit)}
|
||||||
@@ -175,6 +182,7 @@ class Resource(ResourceBase):
|
|||||||
Returns:
|
Returns:
|
||||||
bool -- True if the operation succeeded
|
bool -- True if the operation succeeded
|
||||||
"""
|
"""
|
||||||
|
metrics.track(metrics.COMMIT, self.account, {"name": "delete"})
|
||||||
query = gql(
|
query = gql(
|
||||||
"""
|
"""
|
||||||
mutation CommitDelete($commit: CommitDeleteInput!){ commitDelete(commit: $commit)}
|
mutation CommitDelete($commit: CommitDeleteInput!){ commitDelete(commit: $commit)}
|
||||||
@@ -185,3 +193,41 @@ class Resource(ResourceBase):
|
|||||||
return self.make_request(
|
return self.make_request(
|
||||||
query=query, params=params, return_type="commitDelete", parse_response=False
|
query=query, params=params, return_type="commitDelete", parse_response=False
|
||||||
)
|
)
|
||||||
|
|
||||||
|
def received(
|
||||||
|
self,
|
||||||
|
stream_id: str,
|
||||||
|
commit_id: str,
|
||||||
|
source_application: str = "python",
|
||||||
|
message: Optional[str] = None,
|
||||||
|
) -> bool:
|
||||||
|
"""
|
||||||
|
Mark a commit object a received by the source application.
|
||||||
|
"""
|
||||||
|
metrics.track(metrics.COMMIT, self.account, {"name": "received"})
|
||||||
|
query = gql(
|
||||||
|
"""
|
||||||
|
mutation CommitReceive($receivedInput:CommitReceivedInput!){
|
||||||
|
commitReceive(input:$receivedInput)
|
||||||
|
}
|
||||||
|
"""
|
||||||
|
)
|
||||||
|
params = {
|
||||||
|
"receivedInput": {
|
||||||
|
"sourceApplication": source_application,
|
||||||
|
"streamId": stream_id,
|
||||||
|
"commitId": commit_id,
|
||||||
|
"message": "message",
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
try:
|
||||||
|
return self.make_request(
|
||||||
|
query=query,
|
||||||
|
params=params,
|
||||||
|
return_type="commitReceive",
|
||||||
|
parse_response=False,
|
||||||
|
)
|
||||||
|
except Exception as ex:
|
||||||
|
print(ex.with_traceback)
|
||||||
|
return False
|
||||||
|
|||||||
@@ -1,19 +1,20 @@
|
|||||||
from typing import Dict, List
|
from typing import Dict, List
|
||||||
from gql import gql
|
from gql import gql
|
||||||
from graphql.language import parser
|
|
||||||
from specklepy.api.resource import ResourceBase
|
from specklepy.api.resource import ResourceBase
|
||||||
from specklepy.objects.base import Base
|
from specklepy.objects.base import Base
|
||||||
|
|
||||||
NAME = "object"
|
NAME = "object"
|
||||||
METHODS = []
|
|
||||||
|
|
||||||
|
|
||||||
class Resource(ResourceBase):
|
class Resource(ResourceBase):
|
||||||
"""API Access class for objects"""
|
"""API Access class for objects"""
|
||||||
|
|
||||||
def __init__(self, me, basepath, client) -> None:
|
def __init__(self, account, basepath, client) -> None:
|
||||||
super().__init__(
|
super().__init__(
|
||||||
me=me, basepath=basepath, client=client, name=NAME, methods=METHODS
|
account=account,
|
||||||
|
basepath=basepath,
|
||||||
|
client=client,
|
||||||
|
name=NAME,
|
||||||
)
|
)
|
||||||
self.schema = Base
|
self.schema = Base
|
||||||
|
|
||||||
@@ -49,7 +50,9 @@ class Resource(ResourceBase):
|
|||||||
params = {"stream_id": stream_id, "object_id": object_id}
|
params = {"stream_id": stream_id, "object_id": object_id}
|
||||||
|
|
||||||
return self.make_request(
|
return self.make_request(
|
||||||
query=query, params=params, return_type=["stream", "object", "data"]
|
query=query,
|
||||||
|
params=params,
|
||||||
|
return_type=["stream", "object", "data"],
|
||||||
)
|
)
|
||||||
|
|
||||||
def create(self, stream_id: str, objects: List[Dict]) -> str:
|
def create(self, stream_id: str, objects: List[Dict]) -> str:
|
||||||
|
|||||||
@@ -1,20 +1,24 @@
|
|||||||
from typing import Dict, List
|
import re
|
||||||
|
from typing import Any, Dict, List, Tuple
|
||||||
from gql import gql
|
from gql import gql
|
||||||
from gql.client import Client
|
|
||||||
from specklepy.api.models import ServerInfo
|
from specklepy.api.models import ServerInfo
|
||||||
from specklepy.api.resource import ResourceBase
|
from specklepy.api.resource import ResourceBase
|
||||||
|
from specklepy.logging import metrics
|
||||||
|
from specklepy.logging.exceptions import GraphQLException
|
||||||
|
|
||||||
|
|
||||||
NAME = "server"
|
NAME = "server"
|
||||||
METHODS = ["get", "apps"]
|
|
||||||
|
|
||||||
|
|
||||||
class Resource(ResourceBase):
|
class Resource(ResourceBase):
|
||||||
"""API Access class for the server"""
|
"""API Access class for the server"""
|
||||||
|
|
||||||
def __init__(self, me, basepath, client) -> None:
|
def __init__(self, account, basepath, client) -> None:
|
||||||
super().__init__(
|
super().__init__(
|
||||||
me=me, basepath=basepath, client=client, name=NAME, methods=METHODS
|
account=account,
|
||||||
|
basepath=basepath,
|
||||||
|
client=client,
|
||||||
|
name=NAME,
|
||||||
)
|
)
|
||||||
|
|
||||||
def get(self) -> ServerInfo:
|
def get(self) -> ServerInfo:
|
||||||
@@ -23,6 +27,7 @@ class Resource(ResourceBase):
|
|||||||
Returns:
|
Returns:
|
||||||
dict -- the server info in dictionary form
|
dict -- the server info in dictionary form
|
||||||
"""
|
"""
|
||||||
|
metrics.track(metrics.SERVER, self.account, {"name": "get"})
|
||||||
query = gql(
|
query = gql(
|
||||||
"""
|
"""
|
||||||
query Server {
|
query Server {
|
||||||
@@ -56,12 +61,46 @@ class Resource(ResourceBase):
|
|||||||
query=query, return_type="serverInfo", schema=ServerInfo
|
query=query, return_type="serverInfo", schema=ServerInfo
|
||||||
)
|
)
|
||||||
|
|
||||||
|
def version(self) -> Tuple[Any, ...]:
|
||||||
|
"""Get the server version
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
tuple -- the server version in the format (major, minor, patch, (tag, build))
|
||||||
|
eg (2, 6, 3) for a stable build and (2, 6, 4, 'alpha', 4711) for alpha
|
||||||
|
"""
|
||||||
|
# not tracking as it will be called along with other mutations / queries as a check
|
||||||
|
query = gql(
|
||||||
|
"""
|
||||||
|
query Server {
|
||||||
|
serverInfo {
|
||||||
|
version
|
||||||
|
}
|
||||||
|
}
|
||||||
|
"""
|
||||||
|
)
|
||||||
|
ver = self.make_request(
|
||||||
|
query=query, return_type=["serverInfo", "version"], parse_response=False
|
||||||
|
)
|
||||||
|
if isinstance(ver, Exception):
|
||||||
|
raise GraphQLException(
|
||||||
|
f"Could not get server version for {self.basepath}", [ver]
|
||||||
|
)
|
||||||
|
|
||||||
|
# pylint: disable=consider-using-generator; (list comp is faster)
|
||||||
|
return tuple(
|
||||||
|
[
|
||||||
|
int(segment) if segment.isdigit() else segment
|
||||||
|
for segment in re.split(r"\.|-", ver)
|
||||||
|
]
|
||||||
|
)
|
||||||
|
|
||||||
def apps(self) -> Dict:
|
def apps(self) -> Dict:
|
||||||
"""Get the apps registered on the server
|
"""Get the apps registered on the server
|
||||||
|
|
||||||
Returns:
|
Returns:
|
||||||
dict -- a dictionary of apps registered on the server
|
dict -- a dictionary of apps registered on the server
|
||||||
"""
|
"""
|
||||||
|
metrics.track(metrics.SERVER, self.account, {"name": "apps"})
|
||||||
query = gql(
|
query = gql(
|
||||||
"""
|
"""
|
||||||
query Apps {
|
query Apps {
|
||||||
@@ -95,6 +134,7 @@ class Resource(ResourceBase):
|
|||||||
Returns:
|
Returns:
|
||||||
str -- the new API token. note: this is the only time you'll see the token!
|
str -- the new API token. note: this is the only time you'll see the token!
|
||||||
"""
|
"""
|
||||||
|
metrics.track(metrics.SERVER, self.account, {"name": "create_token"})
|
||||||
query = gql(
|
query = gql(
|
||||||
"""
|
"""
|
||||||
mutation TokenCreate($token: ApiTokenCreateInput!) {
|
mutation TokenCreate($token: ApiTokenCreateInput!) {
|
||||||
@@ -120,6 +160,7 @@ class Resource(ResourceBase):
|
|||||||
Returns:
|
Returns:
|
||||||
bool -- True if the token was successfully deleted
|
bool -- True if the token was successfully deleted
|
||||||
"""
|
"""
|
||||||
|
metrics.track(metrics.SERVER, self.account, {"name": "revoke_token"})
|
||||||
query = gql(
|
query = gql(
|
||||||
"""
|
"""
|
||||||
mutation TokenRevoke($token: String!) {
|
mutation TokenRevoke($token: String!) {
|
||||||
|
|||||||
+515
-102
@@ -1,25 +1,26 @@
|
|||||||
from typing import Dict, List, Optional
|
from datetime import datetime, timezone
|
||||||
|
from typing import List, Optional
|
||||||
|
from deprecated import deprecated
|
||||||
from gql import gql
|
from gql import gql
|
||||||
|
from specklepy.logging import metrics
|
||||||
|
from specklepy.api.models import ActivityCollection, PendingStreamCollaborator, Stream
|
||||||
from specklepy.api.resource import ResourceBase
|
from specklepy.api.resource import ResourceBase
|
||||||
from specklepy.api.models import Stream
|
from specklepy.logging.exceptions import UnsupportedException, SpeckleException
|
||||||
|
|
||||||
|
|
||||||
NAME = "stream"
|
NAME = "stream"
|
||||||
METHODS = [
|
|
||||||
"list",
|
|
||||||
"create",
|
|
||||||
"get",
|
|
||||||
"update",
|
|
||||||
"delete",
|
|
||||||
"search",
|
|
||||||
]
|
|
||||||
|
|
||||||
|
|
||||||
class Resource(ResourceBase):
|
class Resource(ResourceBase):
|
||||||
"""API Access class for streams"""
|
"""API Access class for streams"""
|
||||||
|
|
||||||
def __init__(self, me, basepath, client) -> None:
|
def __init__(self, account, basepath, client, server_version) -> None:
|
||||||
super().__init__(
|
super().__init__(
|
||||||
me=me, basepath=basepath, client=client, name=NAME, methods=METHODS
|
account=account,
|
||||||
|
basepath=basepath,
|
||||||
|
client=client,
|
||||||
|
name=NAME,
|
||||||
|
server_version=server_version,
|
||||||
)
|
)
|
||||||
|
|
||||||
self.schema = Stream
|
self.schema = Stream
|
||||||
@@ -35,44 +36,49 @@ class Resource(ResourceBase):
|
|||||||
Returns:
|
Returns:
|
||||||
Stream -- the retrieved stream
|
Stream -- the retrieved stream
|
||||||
"""
|
"""
|
||||||
|
metrics.track(metrics.STREAM, self.account, {"name": "get"})
|
||||||
query = gql(
|
query = gql(
|
||||||
"""
|
"""
|
||||||
query Stream($id: String!, $branch_limit: Int!, $commit_limit: Int!) {
|
query Stream($id: String!, $branch_limit: Int!, $commit_limit: Int!) {
|
||||||
stream(id: $id) {
|
stream(id: $id) {
|
||||||
id
|
|
||||||
name
|
|
||||||
description
|
|
||||||
isPublic
|
|
||||||
createdAt
|
|
||||||
updatedAt
|
|
||||||
collaborators {
|
|
||||||
id
|
|
||||||
name
|
|
||||||
role
|
|
||||||
avatar
|
|
||||||
}
|
|
||||||
branches(limit: $branch_limit) {
|
|
||||||
totalCount
|
|
||||||
cursor
|
|
||||||
items {
|
|
||||||
id
|
id
|
||||||
name
|
name
|
||||||
|
role
|
||||||
description
|
description
|
||||||
commits(limit: $commit_limit) {
|
isPublic
|
||||||
totalCount
|
createdAt
|
||||||
cursor
|
updatedAt
|
||||||
items {
|
commentCount
|
||||||
|
favoritesCount
|
||||||
|
collaborators {
|
||||||
id
|
id
|
||||||
referencedObject
|
name
|
||||||
message
|
role
|
||||||
authorName
|
avatar
|
||||||
authorId
|
|
||||||
createdAt
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
branches(limit: $branch_limit) {
|
||||||
|
totalCount
|
||||||
|
cursor
|
||||||
|
items {
|
||||||
|
id
|
||||||
|
name
|
||||||
|
description
|
||||||
|
commits(limit: $commit_limit) {
|
||||||
|
totalCount
|
||||||
|
cursor
|
||||||
|
items {
|
||||||
|
id
|
||||||
|
message
|
||||||
|
authorId
|
||||||
|
createdAt
|
||||||
|
authorName
|
||||||
|
referencedObject
|
||||||
|
sourceApplication
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
}
|
}
|
||||||
"""
|
"""
|
||||||
)
|
)
|
||||||
@@ -90,37 +96,41 @@ class Resource(ResourceBase):
|
|||||||
Returns:
|
Returns:
|
||||||
List[Stream] -- A list of Stream objects
|
List[Stream] -- A list of Stream objects
|
||||||
"""
|
"""
|
||||||
|
metrics.track(metrics.STREAM, self.account, {"name": "get"})
|
||||||
query = gql(
|
query = gql(
|
||||||
"""
|
"""
|
||||||
query User($stream_limit: Int!) {
|
query User($stream_limit: Int!) {
|
||||||
user {
|
user {
|
||||||
id
|
|
||||||
email
|
|
||||||
name
|
|
||||||
bio
|
|
||||||
company
|
|
||||||
avatar
|
|
||||||
verified
|
|
||||||
profiles
|
|
||||||
role
|
|
||||||
streams(limit: $stream_limit) {
|
|
||||||
totalCount
|
|
||||||
cursor
|
|
||||||
items {
|
|
||||||
id
|
id
|
||||||
|
bio
|
||||||
name
|
name
|
||||||
description
|
email
|
||||||
isPublic
|
avatar
|
||||||
createdAt
|
company
|
||||||
updatedAt
|
verified
|
||||||
collaborators {
|
profiles
|
||||||
id
|
role
|
||||||
name
|
streams(limit: $stream_limit) {
|
||||||
role
|
totalCount
|
||||||
}
|
cursor
|
||||||
|
items {
|
||||||
|
id
|
||||||
|
name
|
||||||
|
role
|
||||||
|
isPublic
|
||||||
|
createdAt
|
||||||
|
updatedAt
|
||||||
|
description
|
||||||
|
commentCount
|
||||||
|
favoritesCount
|
||||||
|
collaborators {
|
||||||
|
id
|
||||||
|
name
|
||||||
|
role
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
"""
|
"""
|
||||||
)
|
)
|
||||||
@@ -147,6 +157,7 @@ class Resource(ResourceBase):
|
|||||||
Returns:
|
Returns:
|
||||||
id {str} -- the id of the newly created stream
|
id {str} -- the id of the newly created stream
|
||||||
"""
|
"""
|
||||||
|
metrics.track(metrics.STREAM, self.account, {"name": "create"})
|
||||||
query = gql(
|
query = gql(
|
||||||
"""
|
"""
|
||||||
mutation StreamCreate($stream: StreamCreateInput!) {
|
mutation StreamCreate($stream: StreamCreateInput!) {
|
||||||
@@ -177,10 +188,11 @@ class Resource(ResourceBase):
|
|||||||
Returns:
|
Returns:
|
||||||
bool -- whether the stream update was successful
|
bool -- whether the stream update was successful
|
||||||
"""
|
"""
|
||||||
|
metrics.track(metrics.STREAM, self.account, {"name": "update"})
|
||||||
query = gql(
|
query = gql(
|
||||||
"""
|
"""
|
||||||
mutation StreamUpdate($stream: StreamUpdateInput!) {
|
mutation StreamUpdate($stream: StreamUpdateInput!) {
|
||||||
streamUpdate(stream: $stream)
|
streamUpdate(stream: $stream)
|
||||||
}
|
}
|
||||||
"""
|
"""
|
||||||
)
|
)
|
||||||
@@ -207,12 +219,13 @@ class Resource(ResourceBase):
|
|||||||
Returns:
|
Returns:
|
||||||
bool -- whether the deletion was successful
|
bool -- whether the deletion was successful
|
||||||
"""
|
"""
|
||||||
|
metrics.track(metrics.STREAM, self.account, {"name": "delete"})
|
||||||
query = gql(
|
query = gql(
|
||||||
"""
|
"""
|
||||||
mutation StreamDelete($id: String!) {
|
mutation StreamDelete($id: String!) {
|
||||||
streamDelete(id: $id)
|
streamDelete(id: $id)
|
||||||
}
|
}
|
||||||
"""
|
"""
|
||||||
)
|
)
|
||||||
|
|
||||||
params = {"id": id}
|
params = {"id": id}
|
||||||
@@ -239,46 +252,48 @@ class Resource(ResourceBase):
|
|||||||
Returns:
|
Returns:
|
||||||
List[Stream] -- a list of Streams that match the search query
|
List[Stream] -- a list of Streams that match the search query
|
||||||
"""
|
"""
|
||||||
|
metrics.track(metrics.STREAM, self.account, {"name": "search"})
|
||||||
query = gql(
|
query = gql(
|
||||||
"""
|
"""
|
||||||
query StreamSearch($search_query: String!,$limit: Int!, $branch_limit:Int!, $commit_limit:Int!) {
|
query StreamSearch($search_query: String!,$limit: Int!, $branch_limit:Int!, $commit_limit:Int!) {
|
||||||
streams(query: $search_query, limit: $limit) {
|
streams(query: $search_query, limit: $limit) {
|
||||||
items {
|
|
||||||
id
|
|
||||||
name
|
|
||||||
description
|
|
||||||
isPublic
|
|
||||||
createdAt
|
|
||||||
updatedAt
|
|
||||||
collaborators {
|
|
||||||
id
|
|
||||||
name
|
|
||||||
role
|
|
||||||
avatar
|
|
||||||
}
|
|
||||||
branches(limit: $branch_limit) {
|
|
||||||
totalCount
|
|
||||||
cursor
|
|
||||||
items {
|
items {
|
||||||
id
|
id
|
||||||
name
|
name
|
||||||
description
|
role
|
||||||
commits(limit: $commit_limit) {
|
description
|
||||||
totalCount
|
isPublic
|
||||||
cursor
|
createdAt
|
||||||
items {
|
updatedAt
|
||||||
id
|
collaborators {
|
||||||
referencedObject
|
id
|
||||||
message
|
name
|
||||||
authorName
|
role
|
||||||
authorId
|
avatar
|
||||||
createdAt
|
}
|
||||||
|
branches(limit: $branch_limit) {
|
||||||
|
totalCount
|
||||||
|
cursor
|
||||||
|
items {
|
||||||
|
id
|
||||||
|
name
|
||||||
|
description
|
||||||
|
commits(limit: $commit_limit) {
|
||||||
|
totalCount
|
||||||
|
cursor
|
||||||
|
items {
|
||||||
|
id
|
||||||
|
referencedObject
|
||||||
|
message
|
||||||
|
authorName
|
||||||
|
authorId
|
||||||
|
createdAt
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
|
||||||
}
|
}
|
||||||
"""
|
"""
|
||||||
)
|
)
|
||||||
@@ -294,9 +309,51 @@ class Resource(ResourceBase):
|
|||||||
query=query, params=params, return_type=["streams", "items"]
|
query=query, params=params, return_type=["streams", "items"]
|
||||||
)
|
)
|
||||||
|
|
||||||
|
def favorite(self, stream_id: str, favorited: bool = True):
|
||||||
|
"""Favorite or unfavorite the given stream.
|
||||||
|
|
||||||
|
Arguments:
|
||||||
|
stream_id {str} -- the id of the stream to favorite / unfavorite
|
||||||
|
favorited {bool} -- whether to favorite (True) or unfavorite (False) the stream
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Stream -- the stream with its `id`, `name`, and `favoritedDate`
|
||||||
|
"""
|
||||||
|
metrics.track(metrics.STREAM, self.account, {"name": "favorite"})
|
||||||
|
query = gql(
|
||||||
|
"""
|
||||||
|
mutation StreamFavorite($stream_id: String!, $favorited: Boolean!) {
|
||||||
|
streamFavorite(streamId: $stream_id, favorited: $favorited) {
|
||||||
|
id
|
||||||
|
name
|
||||||
|
favoritedDate
|
||||||
|
favoritesCount
|
||||||
|
}
|
||||||
|
}
|
||||||
|
"""
|
||||||
|
)
|
||||||
|
|
||||||
|
params = {
|
||||||
|
"stream_id": stream_id,
|
||||||
|
"favorited": favorited,
|
||||||
|
}
|
||||||
|
|
||||||
|
return self.make_request(
|
||||||
|
query=query, params=params, return_type=["streamFavorite"]
|
||||||
|
)
|
||||||
|
|
||||||
|
@deprecated(
|
||||||
|
version="2.6.4",
|
||||||
|
reason=(
|
||||||
|
"As of Speckle Server v2.6.4, this method is deprecated. "
|
||||||
|
"Users need to be invited and accept the invite before being added to a stream"
|
||||||
|
),
|
||||||
|
)
|
||||||
def grant_permission(self, stream_id: str, user_id: str, role: str):
|
def grant_permission(self, stream_id: str, user_id: str, role: str):
|
||||||
"""Grant permissions to a user on a given stream
|
"""Grant permissions to a user on a given stream
|
||||||
|
|
||||||
|
Valid for Speckle Server version < 2.6.4
|
||||||
|
|
||||||
Arguments:
|
Arguments:
|
||||||
stream_id {str} -- the id of the stream to grant permissions to
|
stream_id {str} -- the id of the stream to grant permissions to
|
||||||
user_id {str} -- the id of the user to grant permissions for
|
user_id {str} -- the id of the user to grant permissions for
|
||||||
@@ -305,6 +362,16 @@ class Resource(ResourceBase):
|
|||||||
Returns:
|
Returns:
|
||||||
bool -- True if the operation was successful
|
bool -- True if the operation was successful
|
||||||
"""
|
"""
|
||||||
|
metrics.track(metrics.PERMISSION, self.account, {"name": "add", "role": role})
|
||||||
|
if self.server_version and self.server_version >= (2, 6, 4):
|
||||||
|
raise UnsupportedException(
|
||||||
|
(
|
||||||
|
"Server mutation `grant_permission` is no longer supported as of Speckle Server v2.6.4. "
|
||||||
|
"Please use the new `update_permission` method to change an existing user's permission "
|
||||||
|
"or use the `invite` method to invite a user to a stream."
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
query = gql(
|
query = gql(
|
||||||
"""
|
"""
|
||||||
mutation StreamGrantPermission($permission_params: StreamGrantPermissionInput !) {
|
mutation StreamGrantPermission($permission_params: StreamGrantPermissionInput !) {
|
||||||
@@ -328,6 +395,282 @@ class Resource(ResourceBase):
|
|||||||
parse_response=False,
|
parse_response=False,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
def get_all_pending_invites(
|
||||||
|
self, stream_id: str
|
||||||
|
) -> List[PendingStreamCollaborator]:
|
||||||
|
"""Get all of the pending invites on a stream.
|
||||||
|
You must be a `stream:owner` to query this.
|
||||||
|
|
||||||
|
Requires Speckle Server version >= 2.6.4
|
||||||
|
|
||||||
|
Arguments:
|
||||||
|
stream_id {str} -- the stream id from which to get the pending invites
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
List[PendingStreamCollaborator] -- a list of pending invites for the specified stream
|
||||||
|
"""
|
||||||
|
metrics.track(metrics.INVITE, self.account, {"name": "get"})
|
||||||
|
self._check_invites_supported()
|
||||||
|
|
||||||
|
query = gql(
|
||||||
|
"""
|
||||||
|
query StreamInvites($streamId: String!) {
|
||||||
|
stream(id: $streamId){
|
||||||
|
pendingCollaborators {
|
||||||
|
id
|
||||||
|
token
|
||||||
|
inviteId
|
||||||
|
streamId
|
||||||
|
streamName
|
||||||
|
title
|
||||||
|
role
|
||||||
|
invitedBy{
|
||||||
|
id
|
||||||
|
name
|
||||||
|
company
|
||||||
|
avatar
|
||||||
|
}
|
||||||
|
user {
|
||||||
|
id
|
||||||
|
name
|
||||||
|
company
|
||||||
|
avatar
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
"""
|
||||||
|
)
|
||||||
|
params = {"streamId": stream_id}
|
||||||
|
|
||||||
|
return self.make_request(
|
||||||
|
query=query,
|
||||||
|
params=params,
|
||||||
|
return_type=["stream", "pendingCollaborators"],
|
||||||
|
schema=PendingStreamCollaborator,
|
||||||
|
)
|
||||||
|
|
||||||
|
def invite(
|
||||||
|
self,
|
||||||
|
stream_id: str,
|
||||||
|
email: str = None,
|
||||||
|
user_id: str = None,
|
||||||
|
role: str = "stream:contributor", # should default be reviewer?
|
||||||
|
message: str = None,
|
||||||
|
):
|
||||||
|
"""Invite someone to a stream using either their email or user id
|
||||||
|
|
||||||
|
Requires Speckle Server version >= 2.6.4
|
||||||
|
|
||||||
|
Arguments:
|
||||||
|
stream_id {str} -- the id of the stream to invite the user to
|
||||||
|
email {str} -- the email of the user to invite (use this OR `user_id`)
|
||||||
|
user_id {str} -- the id of the user to invite (use this OR `email`)
|
||||||
|
role {str} -- the role to assing to the user (defaults to `stream:contributor`)
|
||||||
|
message {str} -- a message to send along with this invite to the specified user
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
bool -- True if the operation was successful
|
||||||
|
"""
|
||||||
|
metrics.track(metrics.INVITE, self.account, {"name": "create"})
|
||||||
|
self._check_invites_supported()
|
||||||
|
|
||||||
|
if email is None and user_id is None:
|
||||||
|
raise SpeckleException(
|
||||||
|
"You must provide either an email or a user id to use the `stream.invite` method"
|
||||||
|
)
|
||||||
|
|
||||||
|
query = gql(
|
||||||
|
"""
|
||||||
|
mutation StreamInviteCreate($input: StreamInviteCreateInput!) {
|
||||||
|
streamInviteCreate(input: $input)
|
||||||
|
}
|
||||||
|
"""
|
||||||
|
)
|
||||||
|
|
||||||
|
params = {
|
||||||
|
"email": email,
|
||||||
|
"userId": user_id,
|
||||||
|
"streamId": stream_id,
|
||||||
|
"message": message,
|
||||||
|
"role": role,
|
||||||
|
}
|
||||||
|
params = {"input": {k: v for k, v in params.items() if v is not None}}
|
||||||
|
|
||||||
|
return self.make_request(
|
||||||
|
query=query,
|
||||||
|
params=params,
|
||||||
|
return_type="streamInviteCreate",
|
||||||
|
parse_response=False,
|
||||||
|
)
|
||||||
|
|
||||||
|
def invite_batch(
|
||||||
|
self,
|
||||||
|
stream_id: str,
|
||||||
|
emails: List[str] = None,
|
||||||
|
user_ids: List[None] = None,
|
||||||
|
message: str = None,
|
||||||
|
) -> bool:
|
||||||
|
"""Invite a batch of users to a specified stream.
|
||||||
|
|
||||||
|
Requires Speckle Server version >= 2.6.4
|
||||||
|
|
||||||
|
Arguments:
|
||||||
|
stream_id {str} -- the id of the stream to invite the user to
|
||||||
|
emails {List[str]} -- the email of the user to invite (use this and/or `user_ids`)
|
||||||
|
user_id {List[str]} -- the id of the user to invite (use this and/or `emails`)
|
||||||
|
message {str} -- a message to send along with this invite to the specified user
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
bool -- True if the operation was successful
|
||||||
|
"""
|
||||||
|
metrics.track(metrics.INVITE, self.account, {"name": "batch create"})
|
||||||
|
self._check_invites_supported()
|
||||||
|
if emails is None and user_ids is None:
|
||||||
|
raise SpeckleException(
|
||||||
|
"You must provide either an email or a user id to use the `stream.invite` method"
|
||||||
|
)
|
||||||
|
|
||||||
|
query = gql(
|
||||||
|
"""
|
||||||
|
mutation StreamInviteBatchCreate($input: [StreamInviteCreateInput!]!) {
|
||||||
|
streamInviteBatchCreate(input: $input)
|
||||||
|
}
|
||||||
|
"""
|
||||||
|
)
|
||||||
|
|
||||||
|
email_invites = [
|
||||||
|
{"streamId": stream_id, "message": message, "email": email}
|
||||||
|
for email in emails
|
||||||
|
if emails is not None
|
||||||
|
]
|
||||||
|
|
||||||
|
user_invites = [
|
||||||
|
{"streamId": stream_id, "message": message, "userId": user_id}
|
||||||
|
for user_id in user_ids
|
||||||
|
if user_ids is not None
|
||||||
|
]
|
||||||
|
|
||||||
|
params = {"input": [*email_invites, *user_invites]}
|
||||||
|
|
||||||
|
return self.make_request(
|
||||||
|
query=query,
|
||||||
|
params=params,
|
||||||
|
return_type="streamInviteBatchCreate",
|
||||||
|
parse_response=False,
|
||||||
|
)
|
||||||
|
|
||||||
|
def invite_cancel(self, stream_id: str, invite_id: str) -> bool:
|
||||||
|
"""Cancel an existing stream invite
|
||||||
|
|
||||||
|
Requires Speckle Server version >= 2.6.4
|
||||||
|
|
||||||
|
Arguments:
|
||||||
|
stream_id {str} -- the id of the stream invite
|
||||||
|
invite_id {str} -- the id of the invite to use
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
bool -- true if the operation was successful
|
||||||
|
"""
|
||||||
|
metrics.track(metrics.INVITE, self.account, {"name": "cancel"})
|
||||||
|
self._check_invites_supported()
|
||||||
|
|
||||||
|
query = gql(
|
||||||
|
"""
|
||||||
|
mutation StreamInviteCancel($streamId: String!, $inviteId: String!) {
|
||||||
|
streamInviteCancel(streamId: $streamId, inviteId: $inviteId)
|
||||||
|
}
|
||||||
|
"""
|
||||||
|
)
|
||||||
|
|
||||||
|
params = {"streamId": stream_id, "inviteId": invite_id}
|
||||||
|
|
||||||
|
return self.make_request(
|
||||||
|
query=query,
|
||||||
|
params=params,
|
||||||
|
return_type="streamInviteCancel",
|
||||||
|
parse_response=False,
|
||||||
|
)
|
||||||
|
|
||||||
|
def invite_use(self, stream_id: str, token: str, accept: bool = True) -> bool:
|
||||||
|
"""Accept or decline a stream invite
|
||||||
|
|
||||||
|
Requires Speckle Server version >= 2.6.4
|
||||||
|
|
||||||
|
Arguments:
|
||||||
|
stream_id {str} -- the id of the stream for which the user has a pending invite
|
||||||
|
token {str} -- the token of the invite to use
|
||||||
|
accept {bool} -- whether or not to accept the invite (defaults to True)
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
bool -- true if the operation was successful
|
||||||
|
"""
|
||||||
|
metrics.track(metrics.INVITE, self.account, {"name": "use"})
|
||||||
|
self._check_invites_supported()
|
||||||
|
|
||||||
|
query = gql(
|
||||||
|
"""
|
||||||
|
mutation StreamInviteUse($accept: Boolean!, $streamId: String!, $token: String!) {
|
||||||
|
streamInviteUse(accept: $accept, streamId: $streamId, token: $token)
|
||||||
|
}
|
||||||
|
"""
|
||||||
|
)
|
||||||
|
|
||||||
|
params = {"streamId": stream_id, "token": token, "accept": accept}
|
||||||
|
|
||||||
|
return self.make_request(
|
||||||
|
query=query,
|
||||||
|
params=params,
|
||||||
|
return_type="streamInviteUse",
|
||||||
|
parse_response=False,
|
||||||
|
)
|
||||||
|
|
||||||
|
def update_permission(self, stream_id: str, user_id: str, role: str):
|
||||||
|
"""Updates permissions for a user on a given stream
|
||||||
|
|
||||||
|
Valid for Speckle Server >=2.6.4
|
||||||
|
|
||||||
|
Arguments:
|
||||||
|
stream_id {str} -- the id of the stream to grant permissions to
|
||||||
|
user_id {str} -- the id of the user to grant permissions for
|
||||||
|
role {str} -- the role to grant the user
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
bool -- True if the operation was successful
|
||||||
|
"""
|
||||||
|
metrics.track(
|
||||||
|
metrics.PERMISSION, self.account, {"name": "update", "role": role}
|
||||||
|
)
|
||||||
|
if self.server_version and self.server_version < (2, 6, 4):
|
||||||
|
raise UnsupportedException(
|
||||||
|
(
|
||||||
|
"Server mutation `update_permission` is only supported as of Speckle Server v2.6.4. "
|
||||||
|
"Please update your Speckle Server to use this method or use the `grant_permission` method instead."
|
||||||
|
)
|
||||||
|
)
|
||||||
|
query = gql(
|
||||||
|
"""
|
||||||
|
mutation StreamUpdatePermission($permission_params: StreamUpdatePermissionInput !) {
|
||||||
|
streamUpdatePermission(permissionParams: $permission_params)
|
||||||
|
}
|
||||||
|
"""
|
||||||
|
)
|
||||||
|
|
||||||
|
params = {
|
||||||
|
"permission_params": {
|
||||||
|
"streamId": stream_id,
|
||||||
|
"userId": user_id,
|
||||||
|
"role": role,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return self.make_request(
|
||||||
|
query=query,
|
||||||
|
params=params,
|
||||||
|
return_type="streamUpdatePermission",
|
||||||
|
parse_response=False,
|
||||||
|
)
|
||||||
|
|
||||||
def revoke_permission(self, stream_id: str, user_id: str):
|
def revoke_permission(self, stream_id: str, user_id: str):
|
||||||
"""Revoke permissions from a user on a given stream
|
"""Revoke permissions from a user on a given stream
|
||||||
|
|
||||||
@@ -338,6 +681,7 @@ class Resource(ResourceBase):
|
|||||||
Returns:
|
Returns:
|
||||||
bool -- True if the operation was successful
|
bool -- True if the operation was successful
|
||||||
"""
|
"""
|
||||||
|
metrics.track(metrics.PERMISSION, self.account, {"name": "revoke"})
|
||||||
query = gql(
|
query = gql(
|
||||||
"""
|
"""
|
||||||
mutation StreamRevokePermission($permission_params: StreamRevokePermissionInput !) {
|
mutation StreamRevokePermission($permission_params: StreamRevokePermissionInput !) {
|
||||||
@@ -354,3 +698,72 @@ class Resource(ResourceBase):
|
|||||||
return_type="streamRevokePermission",
|
return_type="streamRevokePermission",
|
||||||
parse_response=False,
|
parse_response=False,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
def activity(
|
||||||
|
self,
|
||||||
|
stream_id: str,
|
||||||
|
action_type: str = None,
|
||||||
|
limit: int = 20,
|
||||||
|
before: datetime = None,
|
||||||
|
after: datetime = None,
|
||||||
|
cursor: datetime = None,
|
||||||
|
):
|
||||||
|
"""
|
||||||
|
Get the activity from a given stream in an Activity collection. Step into the activity `items` for the list of activity.
|
||||||
|
|
||||||
|
Note: all timestamps arguments should be `datetime` of any tz as they will be converted to UTC ISO format strings
|
||||||
|
|
||||||
|
stream_id {str} -- the id of the stream to get activity from
|
||||||
|
action_type {str} -- filter results to a single action type (eg: `commit_create` or `commit_receive`)
|
||||||
|
limit {int} -- max number of Activity items to return
|
||||||
|
before {datetime} -- latest cutoff for activity (ie: return all activity _before_ this time)
|
||||||
|
after {datetime} -- oldest cutoff for activity (ie: return all activity _after_ this time)
|
||||||
|
cursor {datetime} -- timestamp cursor for pagination
|
||||||
|
"""
|
||||||
|
query = gql(
|
||||||
|
"""
|
||||||
|
query StreamActivity($stream_id: String!, $action_type: String, $before:DateTime, $after: DateTime, $cursor: DateTime, $limit: Int){
|
||||||
|
stream(id: $stream_id) {
|
||||||
|
activity(actionType: $action_type, before: $before, after: $after, cursor: $cursor, limit: $limit) {
|
||||||
|
totalCount
|
||||||
|
cursor
|
||||||
|
items {
|
||||||
|
actionType
|
||||||
|
info
|
||||||
|
userId
|
||||||
|
streamId
|
||||||
|
resourceId
|
||||||
|
resourceType
|
||||||
|
message
|
||||||
|
time
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
"""
|
||||||
|
)
|
||||||
|
try:
|
||||||
|
params = {
|
||||||
|
"stream_id": stream_id,
|
||||||
|
"limit": limit,
|
||||||
|
"action_type": action_type,
|
||||||
|
"before": before.astimezone(timezone.utc).isoformat()
|
||||||
|
if before
|
||||||
|
else before,
|
||||||
|
"after": after.astimezone(timezone.utc).isoformat() if after else after,
|
||||||
|
"cursor": cursor.astimezone(timezone.utc).isoformat()
|
||||||
|
if cursor
|
||||||
|
else cursor,
|
||||||
|
}
|
||||||
|
except AttributeError as e:
|
||||||
|
raise SpeckleException(
|
||||||
|
"Could not get stream activity - `before`, `after`, and `cursor` must be in `datetime` format if provided",
|
||||||
|
ValueError,
|
||||||
|
) from e
|
||||||
|
|
||||||
|
return self.make_request(
|
||||||
|
query=query,
|
||||||
|
params=params,
|
||||||
|
return_type=["stream", "activity"],
|
||||||
|
schema=ActivityCollection,
|
||||||
|
)
|
||||||
|
|||||||
@@ -1,16 +1,12 @@
|
|||||||
from typing import Callable, Dict, List, Optional, Any
|
from typing import Callable, Dict, List, Union
|
||||||
from functools import wraps
|
from functools import wraps
|
||||||
from gql import gql
|
from gql import gql
|
||||||
|
from graphql import DocumentNode
|
||||||
from specklepy.api.resource import ResourceBase
|
from specklepy.api.resource import ResourceBase
|
||||||
from specklepy.api.resources.stream import Stream
|
from specklepy.api.resources.stream import Stream
|
||||||
from specklepy.logging.exceptions import GraphQLException, SpeckleException
|
from specklepy.logging.exceptions import SpeckleException
|
||||||
|
|
||||||
NAME = "subscribe"
|
NAME = "subscribe"
|
||||||
METHODS = [
|
|
||||||
"stream_added",
|
|
||||||
"stream_updated",
|
|
||||||
"stream_removed",
|
|
||||||
]
|
|
||||||
|
|
||||||
|
|
||||||
def check_wsclient(function):
|
def check_wsclient(function):
|
||||||
@@ -29,9 +25,12 @@ def check_wsclient(function):
|
|||||||
class Resource(ResourceBase):
|
class Resource(ResourceBase):
|
||||||
"""API Access class for subscriptions"""
|
"""API Access class for subscriptions"""
|
||||||
|
|
||||||
def __init__(self, me, basepath, client) -> None:
|
def __init__(self, account, basepath, client) -> None:
|
||||||
super().__init__(
|
super().__init__(
|
||||||
me=me, basepath=basepath, client=client, name=NAME, methods=METHODS
|
account=account,
|
||||||
|
basepath=basepath,
|
||||||
|
client=client,
|
||||||
|
name=NAME,
|
||||||
)
|
)
|
||||||
|
|
||||||
@check_wsclient
|
@check_wsclient
|
||||||
@@ -105,15 +104,15 @@ class Resource(ResourceBase):
|
|||||||
@check_wsclient
|
@check_wsclient
|
||||||
async def subscribe(
|
async def subscribe(
|
||||||
self,
|
self,
|
||||||
query: gql,
|
query: DocumentNode,
|
||||||
params: Dict = None,
|
params: Dict = None,
|
||||||
callback: Callable = None,
|
callback: Callable = None,
|
||||||
return_type: str or List = None,
|
return_type: Union[str, List] = None,
|
||||||
schema=None,
|
schema=None,
|
||||||
parse_response: bool = True,
|
parse_response: bool = True,
|
||||||
):
|
):
|
||||||
# if self.client.transport.websocket is None:
|
# if self.client.transport.websocket is None:
|
||||||
# TODO: add multiple subs to the same ws connection
|
# TODO: add multiple subs to the same ws connection
|
||||||
async with self.client as session:
|
async with self.client as session:
|
||||||
async for res in session.subscribe(query, variable_values=params):
|
async for res in session.subscribe(query, variable_values=params):
|
||||||
res = self._step_into_response(response=res, return_type=return_type)
|
res = self._step_into_response(response=res, return_type=return_type)
|
||||||
|
|||||||
@@ -1,20 +1,24 @@
|
|||||||
from specklepy.logging.exceptions import SpeckleException
|
from typing import List, Optional, Union
|
||||||
from typing import List, Optional
|
from datetime import datetime, timezone
|
||||||
from gql import gql
|
from gql import gql
|
||||||
from pydantic.main import BaseModel
|
from specklepy.logging import metrics
|
||||||
|
from specklepy.logging.exceptions import SpeckleException
|
||||||
from specklepy.api.resource import ResourceBase
|
from specklepy.api.resource import ResourceBase
|
||||||
from specklepy.api.models import User
|
from specklepy.api.models import ActivityCollection, PendingStreamCollaborator, User
|
||||||
|
|
||||||
NAME = "user"
|
NAME = "user"
|
||||||
METHODS = ["get"]
|
|
||||||
|
|
||||||
|
|
||||||
class Resource(ResourceBase):
|
class Resource(ResourceBase):
|
||||||
"""API Access class for users"""
|
"""API Access class for users"""
|
||||||
|
|
||||||
def __init__(self, me, basepath, client) -> None:
|
def __init__(self, account, basepath, client, server_version) -> None:
|
||||||
super().__init__(
|
super().__init__(
|
||||||
me=me, basepath=basepath, client=client, name=NAME, methods=METHODS
|
account=account,
|
||||||
|
basepath=basepath,
|
||||||
|
client=client,
|
||||||
|
name=NAME,
|
||||||
|
server_version=server_version,
|
||||||
)
|
)
|
||||||
self.schema = User
|
self.schema = User
|
||||||
|
|
||||||
@@ -27,6 +31,7 @@ class Resource(ResourceBase):
|
|||||||
Returns:
|
Returns:
|
||||||
User -- the retrieved user
|
User -- the retrieved user
|
||||||
"""
|
"""
|
||||||
|
metrics.track(metrics.USER, self.account, {"name": "get"})
|
||||||
query = gql(
|
query = gql(
|
||||||
"""
|
"""
|
||||||
query User($id: String) {
|
query User($id: String) {
|
||||||
@@ -49,7 +54,9 @@ class Resource(ResourceBase):
|
|||||||
|
|
||||||
return self.make_request(query=query, params=params, return_type="user")
|
return self.make_request(query=query, params=params, return_type="user")
|
||||||
|
|
||||||
def search(self, search_query: str, limit: int = 25) -> List[User]:
|
def search(
|
||||||
|
self, search_query: str, limit: int = 25
|
||||||
|
) -> Union[List[User], SpeckleException]:
|
||||||
"""Searches for user by name or email. The search query must be at least 3 characters long
|
"""Searches for user by name or email. The search query must be at least 3 characters long
|
||||||
|
|
||||||
Arguments:
|
Arguments:
|
||||||
@@ -63,6 +70,7 @@ class Resource(ResourceBase):
|
|||||||
message="User search query must be at least 3 characters"
|
message="User search query must be at least 3 characters"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
metrics.track(metrics.USER, self.account, {"name": "search"})
|
||||||
query = gql(
|
query = gql(
|
||||||
"""
|
"""
|
||||||
query UserSearch($search_query: String!, $limit: Int!) {
|
query UserSearch($search_query: String!, $limit: Int!) {
|
||||||
@@ -99,6 +107,7 @@ class Resource(ResourceBase):
|
|||||||
Returns:
|
Returns:
|
||||||
bool -- True if your profile was updated successfully
|
bool -- True if your profile was updated successfully
|
||||||
"""
|
"""
|
||||||
|
metrics.track(metrics.USER, self.account, {"name": "update"})
|
||||||
query = gql(
|
query = gql(
|
||||||
"""
|
"""
|
||||||
mutation UserUpdate($user: UserUpdateInput!) {
|
mutation UserUpdate($user: UserUpdateInput!) {
|
||||||
@@ -118,3 +127,154 @@ class Resource(ResourceBase):
|
|||||||
return self.make_request(
|
return self.make_request(
|
||||||
query=query, params=params, return_type="userUpdate", parse_response=False
|
query=query, params=params, return_type="userUpdate", parse_response=False
|
||||||
)
|
)
|
||||||
|
|
||||||
|
def activity(
|
||||||
|
self,
|
||||||
|
user_id: str = None,
|
||||||
|
limit: int = 20,
|
||||||
|
action_type: str = None,
|
||||||
|
before: datetime = None,
|
||||||
|
after: datetime = None,
|
||||||
|
cursor: datetime = None,
|
||||||
|
):
|
||||||
|
"""
|
||||||
|
Get the activity from a given stream in an Activity collection. Step into the activity `items` for the list of activity.
|
||||||
|
If no id argument is provided, will return the current authenticated user's activity (as extracted from the authorization header).
|
||||||
|
|
||||||
|
Note: all timestamps arguments should be `datetime` of any tz as they will be converted to UTC ISO format strings
|
||||||
|
|
||||||
|
user_id {str} -- the id of the user to get the activity from
|
||||||
|
action_type {str} -- filter results to a single action type (eg: `commit_create` or `commit_receive`)
|
||||||
|
limit {int} -- max number of Activity items to return
|
||||||
|
before {datetime} -- latest cutoff for activity (ie: return all activity _before_ this time)
|
||||||
|
after {datetime} -- oldest cutoff for activity (ie: return all activity _after_ this time)
|
||||||
|
cursor {datetime} -- timestamp cursor for pagination
|
||||||
|
"""
|
||||||
|
|
||||||
|
query = gql(
|
||||||
|
"""
|
||||||
|
query UserActivity($user_id: String, $action_type: String, $before:DateTime, $after: DateTime, $cursor: DateTime, $limit: Int){
|
||||||
|
user(id: $user_id) {
|
||||||
|
activity(actionType: $action_type, before: $before, after: $after, cursor: $cursor, limit: $limit) {
|
||||||
|
totalCount
|
||||||
|
cursor
|
||||||
|
items {
|
||||||
|
actionType
|
||||||
|
info
|
||||||
|
userId
|
||||||
|
streamId
|
||||||
|
resourceId
|
||||||
|
resourceType
|
||||||
|
message
|
||||||
|
time
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
"""
|
||||||
|
)
|
||||||
|
|
||||||
|
params = {
|
||||||
|
"user_id": user_id,
|
||||||
|
"limit": limit,
|
||||||
|
"action_type": action_type,
|
||||||
|
"before": before.astimezone(timezone.utc).isoformat() if before else before,
|
||||||
|
"after": after.astimezone(timezone.utc).isoformat() if after else after,
|
||||||
|
"cursor": cursor.astimezone(timezone.utc).isoformat() if cursor else cursor,
|
||||||
|
}
|
||||||
|
|
||||||
|
return self.make_request(
|
||||||
|
query=query,
|
||||||
|
params=params,
|
||||||
|
return_type=["user", "activity"],
|
||||||
|
schema=ActivityCollection,
|
||||||
|
)
|
||||||
|
|
||||||
|
def get_all_pending_invites(self) -> List[PendingStreamCollaborator]:
|
||||||
|
"""Get all of the active user's pending stream invites
|
||||||
|
|
||||||
|
Requires Speckle Server version >= 2.6.4
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
List[PendingStreamCollaborator] -- a list of pending invites for the current user
|
||||||
|
"""
|
||||||
|
metrics.track(metrics.INVITE, self.account, {"name": "get"})
|
||||||
|
self._check_invites_supported()
|
||||||
|
|
||||||
|
query = gql(
|
||||||
|
"""
|
||||||
|
query StreamInvites {
|
||||||
|
streamInvites{
|
||||||
|
id
|
||||||
|
token
|
||||||
|
inviteId
|
||||||
|
streamId
|
||||||
|
streamName
|
||||||
|
title
|
||||||
|
role
|
||||||
|
invitedBy {
|
||||||
|
id
|
||||||
|
name
|
||||||
|
company
|
||||||
|
avatar
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
"""
|
||||||
|
)
|
||||||
|
|
||||||
|
return self.make_request(
|
||||||
|
query=query,
|
||||||
|
return_type="streamInvites",
|
||||||
|
schema=PendingStreamCollaborator,
|
||||||
|
)
|
||||||
|
|
||||||
|
def get_pending_invite(
|
||||||
|
self, stream_id: str, token: str = None
|
||||||
|
) -> Optional[PendingStreamCollaborator]:
|
||||||
|
"""Get a particular pending invite for the active user on a given stream.
|
||||||
|
If no invite_id is provided, any valid invite will be returned.
|
||||||
|
|
||||||
|
Requires Speckle Server version >= 2.6.4
|
||||||
|
|
||||||
|
Arguments:
|
||||||
|
stream_id {str} -- the id of the stream to look for invites on
|
||||||
|
token {str} -- the token of the invite to look for (optional)
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
PendingStreamCollaborator -- the invite for the given stream (or None if it isn't found)
|
||||||
|
"""
|
||||||
|
metrics.track(metrics.INVITE, self.account, {"name": "get"})
|
||||||
|
self._check_invites_supported()
|
||||||
|
|
||||||
|
query = gql(
|
||||||
|
"""
|
||||||
|
query StreamInvite($streamId: String!, $token: String) {
|
||||||
|
streamInvite(streamId: $streamId, token: $token) {
|
||||||
|
id
|
||||||
|
token
|
||||||
|
streamId
|
||||||
|
streamName
|
||||||
|
title
|
||||||
|
role
|
||||||
|
invitedBy {
|
||||||
|
id
|
||||||
|
name
|
||||||
|
company
|
||||||
|
avatar
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
"""
|
||||||
|
)
|
||||||
|
|
||||||
|
params = {"streamId": stream_id}
|
||||||
|
if token:
|
||||||
|
params["token"] = token
|
||||||
|
|
||||||
|
return self.make_request(
|
||||||
|
query=query,
|
||||||
|
params=params,
|
||||||
|
return_type="streamInvite",
|
||||||
|
schema=PendingStreamCollaborator,
|
||||||
|
)
|
||||||
|
|||||||
@@ -445,7 +445,7 @@ input ServerInfoUpdateInput {
|
|||||||
stream( id: String! ): Stream
|
stream( id: String! ): Stream
|
||||||
|
|
||||||
"""
|
"""
|
||||||
All the streams of the current user, pass in the `query` parameter to seach by name, description or ID.
|
All the streams of the current user, pass in the `query` parameter to search by name, description or ID.
|
||||||
"""
|
"""
|
||||||
streams( query: String, limit: Int = 25, cursor: String ): StreamCollection
|
streams( query: String, limit: Int = 25, cursor: String ): StreamCollection
|
||||||
@hasScope(scope: "streams:read")
|
@hasScope(scope: "streams:read")
|
||||||
|
|||||||
@@ -0,0 +1,166 @@
|
|||||||
|
from warnings import warn
|
||||||
|
from urllib.parse import urlparse, unquote
|
||||||
|
from specklepy.api.credentials import (
|
||||||
|
Account,
|
||||||
|
get_account_from_token,
|
||||||
|
get_local_accounts,
|
||||||
|
)
|
||||||
|
from specklepy.logging import metrics
|
||||||
|
from specklepy.api.client import SpeckleClient
|
||||||
|
from specklepy.transports.server.server import ServerTransport
|
||||||
|
from specklepy.logging.exceptions import SpeckleException, SpeckleWarning
|
||||||
|
|
||||||
|
|
||||||
|
class StreamWrapper:
|
||||||
|
"""
|
||||||
|
The `StreamWrapper` gives you some handy helpers to deal with urls and get authenticated clients and transports.
|
||||||
|
|
||||||
|
Construct a `StreamWrapper` with a stream, branch, commit, or object URL. The corresponding ids will be stored
|
||||||
|
in the wrapper. If you have local accounts on the machine, you can use the `get_account` and `get_client` methods
|
||||||
|
to get a local account for the server. You can also pass a token into `get_client` if you don't have a corresponding
|
||||||
|
local account for the server.
|
||||||
|
|
||||||
|
```py
|
||||||
|
from specklepy.api.wrapper import StreamWrapper
|
||||||
|
|
||||||
|
# provide any stream, branch, commit, object, or globals url
|
||||||
|
wrapper = StreamWrapper("https://speckle.xyz/streams/3073b96e86/commits/604bea8cc6")
|
||||||
|
|
||||||
|
# get an authenticated SpeckleClient if you have a local account for the server
|
||||||
|
client = wrapper.get_client()
|
||||||
|
|
||||||
|
# get an authenticated ServerTransport if you have a local account for the server
|
||||||
|
transport = wrapper.get_transport()
|
||||||
|
```
|
||||||
|
"""
|
||||||
|
|
||||||
|
stream_url: str = None
|
||||||
|
use_ssl: bool = True
|
||||||
|
host: str = None
|
||||||
|
stream_id: str = None
|
||||||
|
commit_id: str = None
|
||||||
|
object_id: str = None
|
||||||
|
branch_name: str = None
|
||||||
|
_client: SpeckleClient = None
|
||||||
|
_account: Account = None
|
||||||
|
|
||||||
|
def __repr__(self):
|
||||||
|
return f"StreamWrapper( server: {self.host}, stream_id: {self.stream_id}, type: {self.type} )"
|
||||||
|
|
||||||
|
def __str__(self) -> str:
|
||||||
|
return self.__repr__()
|
||||||
|
|
||||||
|
@property
|
||||||
|
def type(self) -> str:
|
||||||
|
if self.object_id:
|
||||||
|
return "object"
|
||||||
|
elif self.commit_id:
|
||||||
|
return "commit"
|
||||||
|
elif self.branch_name:
|
||||||
|
return "branch"
|
||||||
|
else:
|
||||||
|
return "stream" if self.stream_id else "invalid"
|
||||||
|
|
||||||
|
def __init__(self, url: str) -> None:
|
||||||
|
self.stream_url = url
|
||||||
|
parsed = urlparse(url)
|
||||||
|
self.host = parsed.netloc
|
||||||
|
self.use_ssl = parsed.scheme == "https"
|
||||||
|
segments = parsed.path.strip("/").split("/", 3)
|
||||||
|
metrics.track(metrics.STREAM_WRAPPER, self.get_account())
|
||||||
|
|
||||||
|
if not segments or len(segments) < 2:
|
||||||
|
raise SpeckleException(
|
||||||
|
f"Cannot parse {url} into a stream wrapper class - invalid URL provided."
|
||||||
|
)
|
||||||
|
|
||||||
|
while segments:
|
||||||
|
segment = segments.pop(0)
|
||||||
|
if segments and segment.lower() == "streams":
|
||||||
|
self.stream_id = segments.pop(0)
|
||||||
|
elif segments and segment.lower() == "commits":
|
||||||
|
self.commit_id = segments.pop(0)
|
||||||
|
elif segments and segment.lower() == "branches":
|
||||||
|
self.branch_name = unquote(segments.pop(0))
|
||||||
|
elif segments and segment.lower() == "objects":
|
||||||
|
self.object_id = segments.pop(0)
|
||||||
|
elif segment.lower() == "globals":
|
||||||
|
self.branch_name = "globals"
|
||||||
|
if segments:
|
||||||
|
self.commit_id = segments.pop(0)
|
||||||
|
else:
|
||||||
|
raise SpeckleException(
|
||||||
|
f"Cannot parse {url} into a stream wrapper class - invalid URL provided."
|
||||||
|
)
|
||||||
|
|
||||||
|
if not self.stream_id:
|
||||||
|
raise SpeckleException(
|
||||||
|
f"Cannot parse {url} into a stream wrapper class - no stream id found."
|
||||||
|
)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def server_url(self):
|
||||||
|
return f"{'https' if self.use_ssl else 'http'}://{self.host}"
|
||||||
|
|
||||||
|
def get_account(self, token: str = None) -> Account:
|
||||||
|
"""
|
||||||
|
Gets an account object for this server from the local accounts db (added via Speckle Manager or a json file)
|
||||||
|
"""
|
||||||
|
if self._account and self._account.token:
|
||||||
|
return self._account
|
||||||
|
|
||||||
|
self._account = next(
|
||||||
|
(a for a in get_local_accounts() if self.host in a.serverInfo.url),
|
||||||
|
None,
|
||||||
|
)
|
||||||
|
|
||||||
|
if not self._account:
|
||||||
|
self._account = get_account_from_token(token, self.server_url)
|
||||||
|
|
||||||
|
if self._client:
|
||||||
|
self._client.authenticate_with_account(self._account)
|
||||||
|
|
||||||
|
return self._account
|
||||||
|
|
||||||
|
def get_client(self, token: str = None) -> SpeckleClient:
|
||||||
|
"""
|
||||||
|
Gets an authenticated client for this server. You may provide a token if there aren't any local accounts on this
|
||||||
|
machine. If no account is found and no token is provided, an unauthenticated client is returned.
|
||||||
|
|
||||||
|
Arguments:
|
||||||
|
token {str} -- optional token if no local account is available (defaults to None)
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
SpeckleClient -- authenticated with a corresponding local account or the provided token
|
||||||
|
"""
|
||||||
|
if self._client and token is None:
|
||||||
|
return self._client
|
||||||
|
|
||||||
|
if not self._account or not self._account.token:
|
||||||
|
self.get_account(token)
|
||||||
|
|
||||||
|
if not self._client:
|
||||||
|
self._client = SpeckleClient(host=self.host, use_ssl=self.use_ssl)
|
||||||
|
|
||||||
|
if self._account.token is None and token is None:
|
||||||
|
warn(f"No local account found for server {self.host}", SpeckleWarning)
|
||||||
|
return self._client
|
||||||
|
|
||||||
|
if self._account.token:
|
||||||
|
self._client.authenticate_with_account(self._account)
|
||||||
|
else:
|
||||||
|
self._client.authenticate_with_token(token)
|
||||||
|
|
||||||
|
return self._client
|
||||||
|
|
||||||
|
def get_transport(self, token: str = None) -> ServerTransport:
|
||||||
|
"""
|
||||||
|
Gets a server transport for this stream using an authenticated client. If there is no local account for this
|
||||||
|
server and the client was not authenticated with a token, this will throw an exception.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
ServerTransport -- constructed for this stream with a pre-authenticated client
|
||||||
|
"""
|
||||||
|
if not self._account or not self._account.token:
|
||||||
|
self.get_account(token)
|
||||||
|
return ServerTransport(self.stream_id, account=self._account)
|
||||||
@@ -1,8 +1,9 @@
|
|||||||
from typing import Any, List
|
from typing import Any, List, Optional
|
||||||
|
|
||||||
|
|
||||||
class SpeckleException(Exception):
|
class SpeckleException(Exception):
|
||||||
def __init__(self, message: str, exception: Exception = None) -> None:
|
def __init__(self, message: str, exception: Exception = None) -> None:
|
||||||
|
super().__init__()
|
||||||
self.message = message
|
self.message = message
|
||||||
self.exception = exception
|
self.exception = exception
|
||||||
|
|
||||||
@@ -11,17 +12,19 @@ class SpeckleException(Exception):
|
|||||||
|
|
||||||
|
|
||||||
class SerializationException(SpeckleException):
|
class SerializationException(SpeckleException):
|
||||||
def __init__(self, message: str, object: Any, exception: Exception = None) -> None:
|
def __init__(self, message: str, obj: Any, exception: Exception = None) -> None:
|
||||||
super().__init__(message=message)
|
super().__init__(message=message, exception=exception)
|
||||||
self.object = object
|
self.obj = obj
|
||||||
self.unhandled_type = type(object)
|
self.unhandled_type = type(obj)
|
||||||
|
|
||||||
def __str__(self) -> str:
|
def __str__(self) -> str:
|
||||||
return f"SpeckleException: Could not serialize object of type {self.unhandled_type}"
|
return f"SpeckleException: Could not serialize object of type {self.unhandled_type}"
|
||||||
|
|
||||||
|
|
||||||
class GraphQLException(SpeckleException):
|
class GraphQLException(SpeckleException):
|
||||||
def __init__(self, message: str, errors: List, data=None) -> None:
|
def __init__(
|
||||||
|
self, message: str, errors: Optional[List[Any]] = None, data=None
|
||||||
|
) -> None:
|
||||||
super().__init__(message=message)
|
super().__init__(message=message)
|
||||||
self.errors = errors
|
self.errors = errors
|
||||||
self.data = data
|
self.data = data
|
||||||
@@ -30,6 +33,14 @@ class GraphQLException(SpeckleException):
|
|||||||
return f"GraphQLException: {self.message}"
|
return f"GraphQLException: {self.message}"
|
||||||
|
|
||||||
|
|
||||||
|
class UnsupportedException(SpeckleException):
|
||||||
|
def __init__(self, message: str) -> None:
|
||||||
|
super().__init__(message=message)
|
||||||
|
|
||||||
|
def __str__(self) -> str:
|
||||||
|
return f"UnsupportedException: {self.message}"
|
||||||
|
|
||||||
|
|
||||||
class SpeckleWarning(Warning):
|
class SpeckleWarning(Warning):
|
||||||
def __init__(self, *args: object) -> None:
|
def __init__(self, *args: object) -> None:
|
||||||
super().__init__(*args)
|
super().__init__(*args)
|
||||||
|
|||||||
@@ -0,0 +1,148 @@
|
|||||||
|
import sys
|
||||||
|
import queue
|
||||||
|
import hashlib
|
||||||
|
import getpass
|
||||||
|
import logging
|
||||||
|
import requests
|
||||||
|
import threading
|
||||||
|
import platform
|
||||||
|
import contextlib
|
||||||
|
|
||||||
|
|
||||||
|
"""
|
||||||
|
Anonymous telemetry to help us understand how to make a better Speckle.
|
||||||
|
This really helps us to deliver a better open source project and product!
|
||||||
|
"""
|
||||||
|
TRACK = True
|
||||||
|
HOST_APP = "python"
|
||||||
|
HOST_APP_VERSION = f"python {'.'.join(map(str, sys.version_info[:2]))}"
|
||||||
|
PLATFORMS = {"win32": "Windows", "cygwin": "Windows", "darwin": "Mac OS X"}
|
||||||
|
|
||||||
|
LOG = logging.getLogger(__name__)
|
||||||
|
METRICS_TRACKER = None
|
||||||
|
|
||||||
|
# actions
|
||||||
|
RECEIVE = "Receive"
|
||||||
|
SEND = "Send"
|
||||||
|
STREAM = "Stream Action"
|
||||||
|
PERMISSION = "Permission Action"
|
||||||
|
INVITE = "Invite Action"
|
||||||
|
COMMIT = "Commit Action"
|
||||||
|
BRANCH = "Branch Action"
|
||||||
|
USER = "User Action"
|
||||||
|
SERVER = "Server Action"
|
||||||
|
CLIENT = "Speckle Client"
|
||||||
|
STREAM_WRAPPER = "Stream Wrapper"
|
||||||
|
|
||||||
|
ACCOUNTS = "Get Local Accounts"
|
||||||
|
|
||||||
|
SERIALIZE = "serialization/serialize"
|
||||||
|
DESERIALIZE = "serialization/deserialize"
|
||||||
|
|
||||||
|
|
||||||
|
def disable():
|
||||||
|
global TRACK
|
||||||
|
TRACK = False
|
||||||
|
|
||||||
|
|
||||||
|
def enable():
|
||||||
|
global TRACK
|
||||||
|
TRACK = True
|
||||||
|
|
||||||
|
|
||||||
|
def set_host_app(host_app: str, host_app_version: str = None):
|
||||||
|
global HOST_APP, HOST_APP_VERSION
|
||||||
|
HOST_APP = host_app
|
||||||
|
HOST_APP_VERSION = host_app_version or HOST_APP_VERSION
|
||||||
|
|
||||||
|
|
||||||
|
def track(action: str, account: "Account" = None, custom_props: dict = None):
|
||||||
|
if not TRACK:
|
||||||
|
return
|
||||||
|
try:
|
||||||
|
initialise_tracker(account)
|
||||||
|
event_params = {
|
||||||
|
"event": action,
|
||||||
|
"properties": {
|
||||||
|
"distinct_id": METRICS_TRACKER.last_user,
|
||||||
|
"server_id": METRICS_TRACKER.last_server,
|
||||||
|
"token": METRICS_TRACKER.analytics_token,
|
||||||
|
"hostApp": HOST_APP,
|
||||||
|
"hostAppVersion": HOST_APP_VERSION,
|
||||||
|
"$os": METRICS_TRACKER.platform,
|
||||||
|
"type": "action",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
if custom_props:
|
||||||
|
event_params["properties"].update(custom_props)
|
||||||
|
|
||||||
|
METRICS_TRACKER.queue.put_nowait(event_params)
|
||||||
|
except Exception as ex:
|
||||||
|
# wrapping this whole thing in a try except as we never want a failure here to annoy users!
|
||||||
|
LOG.error(f"Error queueing metrics request: {str(ex)}")
|
||||||
|
|
||||||
|
|
||||||
|
def initialise_tracker(account: "Account" = None):
|
||||||
|
global METRICS_TRACKER
|
||||||
|
if not METRICS_TRACKER:
|
||||||
|
METRICS_TRACKER = MetricsTracker()
|
||||||
|
|
||||||
|
if account and account.userInfo.email:
|
||||||
|
METRICS_TRACKER.set_last_user(account.userInfo.email)
|
||||||
|
if account and account.serverInfo.url:
|
||||||
|
METRICS_TRACKER.set_last_server(account.userInfo.email)
|
||||||
|
|
||||||
|
|
||||||
|
class Singleton(type):
|
||||||
|
_instances = {}
|
||||||
|
|
||||||
|
def __call__(cls, *args, **kwargs):
|
||||||
|
if cls not in cls._instances:
|
||||||
|
cls._instances[cls] = super(Singleton, cls).__call__(*args, **kwargs)
|
||||||
|
return cls._instances[cls]
|
||||||
|
|
||||||
|
|
||||||
|
class MetricsTracker(metaclass=Singleton):
|
||||||
|
analytics_url = "https://analytics.speckle.systems/track?ip=1"
|
||||||
|
analytics_token = "acd87c5a50b56df91a795e999812a3a4"
|
||||||
|
last_user = ""
|
||||||
|
last_server = None
|
||||||
|
platform = None
|
||||||
|
sending_thread = None
|
||||||
|
queue = queue.Queue(1000)
|
||||||
|
|
||||||
|
def __init__(self) -> None:
|
||||||
|
self.sending_thread = threading.Thread(
|
||||||
|
target=self._send_tracking_requests, daemon=True
|
||||||
|
)
|
||||||
|
self.platform = PLATFORMS.get(sys.platform, "linux")
|
||||||
|
self.sending_thread.start()
|
||||||
|
with contextlib.suppress(Exception):
|
||||||
|
node, user = platform.node(), getpass.getuser()
|
||||||
|
if node and user:
|
||||||
|
self.last_user = f"@{self.hash(f'{node}-{user}')}"
|
||||||
|
|
||||||
|
def set_last_user(self, email: str):
|
||||||
|
if not email:
|
||||||
|
return
|
||||||
|
self.last_user = f"@{self.hash(email)}"
|
||||||
|
|
||||||
|
def set_last_server(self, server: str):
|
||||||
|
if not server:
|
||||||
|
return
|
||||||
|
self.last_server = self.hash(server)
|
||||||
|
|
||||||
|
def hash(self, value: str):
|
||||||
|
return hashlib.md5(value.lower().encode("utf-8")).hexdigest().upper()
|
||||||
|
|
||||||
|
def _send_tracking_requests(self):
|
||||||
|
session = requests.Session()
|
||||||
|
while True:
|
||||||
|
event_params = [self.queue.get()]
|
||||||
|
|
||||||
|
try:
|
||||||
|
session.post(self.analytics_url, json=event_params)
|
||||||
|
except Exception as ex:
|
||||||
|
LOG.error(f"Error sending metrics request: {str(ex)}")
|
||||||
|
|
||||||
|
self.queue.task_done()
|
||||||
+58
-33
@@ -1,6 +1,17 @@
|
|||||||
import typing
|
from typing import (
|
||||||
from typing import (Any, Callable, ClassVar, Dict, List, Optional, Set, Type,
|
Any,
|
||||||
get_type_hints)
|
ClassVar,
|
||||||
|
Dict,
|
||||||
|
List,
|
||||||
|
Optional,
|
||||||
|
Union,
|
||||||
|
Set,
|
||||||
|
Type,
|
||||||
|
get_type_hints,
|
||||||
|
)
|
||||||
|
|
||||||
|
import contextlib
|
||||||
|
from enum import EnumMeta
|
||||||
from warnings import warn
|
from warnings import warn
|
||||||
|
|
||||||
from specklepy.logging.exceptions import SpeckleException
|
from specklepy.logging.exceptions import SpeckleException
|
||||||
@@ -81,12 +92,19 @@ class _RegisteringBase:
|
|||||||
speckle_type: ClassVar[str]
|
speckle_type: ClassVar[str]
|
||||||
_type_registry: ClassVar[Dict[str, "Base"]] = {}
|
_type_registry: ClassVar[Dict[str, "Base"]] = {}
|
||||||
_attr_types: ClassVar[Dict[str, Type]] = {}
|
_attr_types: ClassVar[Dict[str, Type]] = {}
|
||||||
|
# dict of chunkable props and their max chunk size
|
||||||
|
_chunkable: Dict[str, int] = {}
|
||||||
|
_chunk_size_default: int = 1000
|
||||||
|
_detachable: Set[str] = set() # list of defined detachable props
|
||||||
|
_serialize_ignore: Set[str] = set()
|
||||||
|
|
||||||
class Config:
|
class Config:
|
||||||
validate_assignment = True
|
validate_assignment = True
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def get_registered_type(cls, speckle_type: str) -> Optional[Type["Base"]]:
|
def get_registered_type(
|
||||||
|
cls, speckle_type: str
|
||||||
|
) -> Union["Base", Type["Base"], None]:
|
||||||
"""Get the registered type from the protected mapping via the `speckle_type`"""
|
"""Get the registered type from the protected mapping via the `speckle_type`"""
|
||||||
return cls._type_registry.get(speckle_type, None)
|
return cls._type_registry.get(speckle_type, None)
|
||||||
|
|
||||||
@@ -118,14 +136,12 @@ class _RegisteringBase:
|
|||||||
except Exception:
|
except Exception:
|
||||||
cls._attr_types = getattr(cls, "__annotations__", {})
|
cls._attr_types = getattr(cls, "__annotations__", {})
|
||||||
if chunkable:
|
if chunkable:
|
||||||
chunkable = {k: v for k, v in chunkable.items()
|
chunkable = {k: v for k, v in chunkable.items() if isinstance(v, int)}
|
||||||
if isinstance(v, int)}
|
|
||||||
cls._chunkable = dict(cls._chunkable, **chunkable)
|
cls._chunkable = dict(cls._chunkable, **chunkable)
|
||||||
if detachable:
|
if detachable:
|
||||||
cls._detachable = cls._detachable.union(detachable)
|
cls._detachable = cls._detachable.union(detachable)
|
||||||
if serialize_ignore:
|
if serialize_ignore:
|
||||||
cls._serialize_ignore = cls._serialize_ignore.union(
|
cls._serialize_ignore = cls._serialize_ignore.union(serialize_ignore)
|
||||||
serialize_ignore)
|
|
||||||
super().__init_subclass__(**kwargs)
|
super().__init_subclass__(**kwargs)
|
||||||
|
|
||||||
|
|
||||||
@@ -133,12 +149,7 @@ class Base(_RegisteringBase):
|
|||||||
id: Optional[str] = None
|
id: Optional[str] = None
|
||||||
totalChildrenCount: Optional[int] = None
|
totalChildrenCount: Optional[int] = None
|
||||||
applicationId: Optional[str] = None
|
applicationId: Optional[str] = None
|
||||||
_units: str = "m"
|
_units: Union[str, None] = None
|
||||||
# dict of chunkable props and their max chunk size
|
|
||||||
_chunkable: Dict[str, int] = {}
|
|
||||||
_chunk_size_default: int = 1000
|
|
||||||
_detachable: Set[str] = set() # list of defined detachable props
|
|
||||||
_serialize_ignore: Set[str] = set()
|
|
||||||
|
|
||||||
def __init__(self, **kwargs) -> None:
|
def __init__(self, **kwargs) -> None:
|
||||||
super().__init__()
|
super().__init__()
|
||||||
@@ -200,7 +211,7 @@ class Base(_RegisteringBase):
|
|||||||
try:
|
try:
|
||||||
attr.__set__(self, value)
|
attr.__set__(self, value)
|
||||||
except AttributeError:
|
except AttributeError:
|
||||||
pass # the prop probably doesn't have a setter
|
return # the prop probably doesn't have a setter
|
||||||
super().__setattr__(name, value)
|
super().__setattr__(name, value)
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
@@ -215,15 +226,13 @@ class Base(_RegisteringBase):
|
|||||||
try:
|
try:
|
||||||
cls._attr_types = get_type_hints(cls)
|
cls._attr_types = get_type_hints(cls)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
warn(
|
warn(f"Could not update forward refs for class {cls.__name__}: {e}")
|
||||||
f"Could not update forward refs for class {cls.__name__}: {e}")
|
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def validate_prop_name(cls, name: str) -> None:
|
def validate_prop_name(cls, name: str) -> None:
|
||||||
"""Validator for dynamic attribute names."""
|
"""Validator for dynamic attribute names."""
|
||||||
if name in {"", "@"}:
|
if name in {"", "@"}:
|
||||||
raise ValueError(
|
raise ValueError("Invalid Name: Base member names cannot be empty strings")
|
||||||
"Invalid Name: Base member names cannot be empty strings")
|
|
||||||
if name.startswith("@@"):
|
if name.startswith("@@"):
|
||||||
raise ValueError(
|
raise ValueError(
|
||||||
"Invalid Name: Base member names cannot start with more than one '@'",
|
"Invalid Name: Base member names cannot start with more than one '@'",
|
||||||
@@ -244,12 +253,23 @@ class Base(_RegisteringBase):
|
|||||||
types = getattr(self, "_attr_types", {})
|
types = getattr(self, "_attr_types", {})
|
||||||
t = types.get(name, None)
|
t = types.get(name, None)
|
||||||
|
|
||||||
if t is None:
|
if t is None or t is Any:
|
||||||
return value
|
return value
|
||||||
|
|
||||||
|
if value is None:
|
||||||
|
return None
|
||||||
|
|
||||||
|
if isinstance(t, EnumMeta) and (value in t._value2member_map_):
|
||||||
|
return t(value)
|
||||||
|
|
||||||
if t.__module__ == "typing":
|
if t.__module__ == "typing":
|
||||||
origin = getattr(t, "__origin__")
|
origin = getattr(t, "__origin__")
|
||||||
t = t.__args__ if origin is typing.Union else origin
|
t = (
|
||||||
|
tuple(getattr(sub_t, "__origin__", sub_t) for sub_t in t.__args__)
|
||||||
|
if origin is Union
|
||||||
|
else origin
|
||||||
|
)
|
||||||
|
|
||||||
if not isinstance(t, (type, tuple)):
|
if not isinstance(t, (type, tuple)):
|
||||||
warn(
|
warn(
|
||||||
f"Unrecognised type '{t}' provided for attribute '{name}'. Type will not been validated."
|
f"Unrecognised type '{t}' provided for attribute '{name}'. Type will not been validated."
|
||||||
@@ -260,13 +280,14 @@ class Base(_RegisteringBase):
|
|||||||
|
|
||||||
# to be friendly, we'll parse ints and strs into floats, but not the other way around
|
# to be friendly, we'll parse ints and strs into floats, but not the other way around
|
||||||
# (to avoid unexpected rounding)
|
# (to avoid unexpected rounding)
|
||||||
if t is float and isinstance(value, (int, str, float)):
|
if isinstance(t, tuple):
|
||||||
try:
|
t = t[0]
|
||||||
|
|
||||||
|
with contextlib.suppress(ValueError):
|
||||||
|
if t is float:
|
||||||
return float(value)
|
return float(value)
|
||||||
except ValueError:
|
if t is str and value:
|
||||||
pass
|
return str(value)
|
||||||
if t is str and value is not None:
|
|
||||||
return str(value)
|
|
||||||
|
|
||||||
raise SpeckleException(
|
raise SpeckleException(
|
||||||
f"Cannot set '{self.__class__.__name__}.{name}': it expects type '{t.__name__}', but received type '{type(value).__name__}'"
|
f"Cannot set '{self.__class__.__name__}.{name}': it expects type '{t.__name__}', but received type '{type(value).__name__}'"
|
||||||
@@ -297,7 +318,9 @@ class Base(_RegisteringBase):
|
|||||||
|
|
||||||
@units.setter
|
@units.setter
|
||||||
def units(self, value: str):
|
def units(self, value: str):
|
||||||
self._units = get_units_from_string(value)
|
units = get_units_from_string(value)
|
||||||
|
if units:
|
||||||
|
self._units = units
|
||||||
|
|
||||||
def get_member_names(self) -> List[str]:
|
def get_member_names(self) -> List[str]:
|
||||||
"""Get all of the property names on this object, dynamic or not"""
|
"""Get all of the property names on this object, dynamic or not"""
|
||||||
@@ -310,7 +333,7 @@ class Base(_RegisteringBase):
|
|||||||
|
|
||||||
def get_serializable_attributes(self) -> List[str]:
|
def get_serializable_attributes(self) -> List[str]:
|
||||||
"""Get the attributes that should be serialized"""
|
"""Get the attributes that should be serialized"""
|
||||||
return list(set(self.get_member_names()) - self._serialize_ignore)
|
return sorted(list(set(self.get_member_names()) - self._serialize_ignore))
|
||||||
|
|
||||||
def get_typed_member_names(self) -> List[str]:
|
def get_typed_member_names(self) -> List[str]:
|
||||||
"""Get all of the names of the defined (typed) properties of this object"""
|
"""Get all of the names of the defined (typed) properties of this object"""
|
||||||
@@ -327,7 +350,8 @@ class Base(_RegisteringBase):
|
|||||||
|
|
||||||
def get_id(self, decompose: bool = False) -> str:
|
def get_id(self, decompose: bool = False) -> str:
|
||||||
"""
|
"""
|
||||||
Gets the id (a unique hash) of this object. ⚠️ This method fully serializes the object, which in the case of large objects (with many sub-objects), has a tangible cost. Avoid using it!
|
Gets the id (a unique hash) of this object. ⚠️ This method fully serializes the object which,
|
||||||
|
in the case of large objects (with many sub-objects), has a tangible cost. Avoid using it!
|
||||||
|
|
||||||
Note: the hash of a decomposed object differs from that of a non-decomposed object
|
Note: the hash of a decomposed object differs from that of a non-decomposed object
|
||||||
|
|
||||||
@@ -337,8 +361,7 @@ class Base(_RegisteringBase):
|
|||||||
Returns:
|
Returns:
|
||||||
str -- the hash (id) of the fully serialized object
|
str -- the hash (id) of the fully serialized object
|
||||||
"""
|
"""
|
||||||
from specklepy.serialization.base_object_serializer import \
|
from specklepy.serialization.base_object_serializer import BaseObjectSerializer
|
||||||
BaseObjectSerializer
|
|
||||||
|
|
||||||
serializer = BaseObjectSerializer()
|
serializer = BaseObjectSerializer()
|
||||||
if decompose:
|
if decompose:
|
||||||
@@ -357,6 +380,7 @@ class Base(_RegisteringBase):
|
|||||||
)
|
)
|
||||||
|
|
||||||
def _handle_object_count(self, obj: Any, parsed: List) -> int:
|
def _handle_object_count(self, obj: Any, parsed: List) -> int:
|
||||||
|
# pylint: disable=isinstance-second-argument-not-valid-type
|
||||||
count = 0
|
count = 0
|
||||||
if obj is None:
|
if obj is None:
|
||||||
return count
|
return count
|
||||||
@@ -385,7 +409,8 @@ Base.update_forward_refs()
|
|||||||
|
|
||||||
|
|
||||||
class DataChunk(Base, speckle_type="Speckle.Core.Models.DataChunk"):
|
class DataChunk(Base, speckle_type="Speckle.Core.Models.DataChunk"):
|
||||||
data: List[Any] = None
|
data: Union[List[Any], None] = None
|
||||||
|
|
||||||
def __init__(self) -> None:
|
def __init__(self) -> None:
|
||||||
|
super().__init__()
|
||||||
self.data = []
|
self.data = []
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
from enum import Enum
|
from enum import Enum
|
||||||
from typing import Any, Callable, List, Type
|
from typing import Any, Callable, List, Type, Dict
|
||||||
|
|
||||||
from specklepy.logging.exceptions import SpeckleException
|
from specklepy.logging.exceptions import SpeckleException
|
||||||
from specklepy.objects.base import Base
|
from specklepy.objects.base import Base
|
||||||
@@ -17,6 +17,7 @@ class CurveTypeEncoding(int, Enum):
|
|||||||
@property
|
@property
|
||||||
def object_class(self) -> Type:
|
def object_class(self) -> Type:
|
||||||
from . import geometry
|
from . import geometry
|
||||||
|
|
||||||
if self == self.Arc:
|
if self == self.Arc:
|
||||||
return geometry.Arc
|
return geometry.Arc
|
||||||
elif self == self.Circle:
|
elif self == self.Circle:
|
||||||
@@ -32,7 +33,8 @@ class CurveTypeEncoding(int, Enum):
|
|||||||
elif self == self.Polycurve:
|
elif self == self.Polycurve:
|
||||||
return geometry.Polycurve
|
return geometry.Polycurve
|
||||||
raise SpeckleException(
|
raise SpeckleException(
|
||||||
f'No corresponding object class for CurveTypeEncoding: {self}')
|
f"No corresponding object class for CurveTypeEncoding: {self}"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
def curve_from_list(args: List[float]):
|
def curve_from_list(args: List[float]):
|
||||||
@@ -41,61 +43,64 @@ def curve_from_list(args: List[float]):
|
|||||||
|
|
||||||
|
|
||||||
class ObjectArray:
|
class ObjectArray:
|
||||||
|
def __init__(self, data: list = None) -> None:
|
||||||
def __init__(self) -> None:
|
self.data = data or []
|
||||||
self.data = []
|
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def from_objects(cls, objects: List[Base]) -> 'ObjectArray':
|
def from_objects(cls, objects: List[Base]) -> "ObjectArray":
|
||||||
data_chunk = cls()
|
data_list = cls()
|
||||||
if len(objects) == 0:
|
if not objects:
|
||||||
return data_chunk
|
return data_list
|
||||||
|
|
||||||
speckle_type = objects[0].speckle_type
|
speckle_type = objects[0].speckle_type
|
||||||
|
|
||||||
for obj in objects:
|
for obj in objects:
|
||||||
if speckle_type != obj.speckle_type:
|
if speckle_type != obj.speckle_type:
|
||||||
raise SpeckleException(
|
raise SpeckleException(
|
||||||
'All objects in chunk should have the same speckle_type. '
|
"All objects in chunk should have the same speckle_type. "
|
||||||
f'Found {speckle_type} and {obj.speckle_type}'
|
f"Found {speckle_type} and {obj.speckle_type}"
|
||||||
)
|
)
|
||||||
data_chunk.encode_object(object=obj)
|
data_list.encode_object(obj=obj)
|
||||||
|
|
||||||
return data_chunk
|
return data_list
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def decode_data(data: List[Any], decoder: Callable[[List[Any]], Base]) -> List[Base]:
|
def decode_data(
|
||||||
|
data: List[Any], decoder: Callable[[List[Any]], Base], **kwargs: Dict[str, Any]
|
||||||
|
) -> List[Base]:
|
||||||
|
bases = []
|
||||||
|
if not data:
|
||||||
|
return bases
|
||||||
index = 0
|
index = 0
|
||||||
unchunked_data = []
|
|
||||||
while index < len(data):
|
while index < len(data):
|
||||||
chunk_length = data[index]
|
item_length = int(data[index])
|
||||||
chunk_start = int(index + 1)
|
item_start = index + 1
|
||||||
chunk_end = int(chunk_start + chunk_length)
|
item_end = item_start + item_length
|
||||||
chunk_data = data[chunk_start:chunk_end]
|
item_data = data[item_start:item_end]
|
||||||
decoded_data = decoder(chunk_data)
|
index = item_end
|
||||||
unchunked_data.append(decoded_data)
|
decoded_data = decoder(item_data, **kwargs)
|
||||||
index = chunk_end
|
bases.append(decoded_data)
|
||||||
return unchunked_data
|
|
||||||
|
|
||||||
def decode(self, decoder: Callable[[List[Any]], Any]):
|
return bases
|
||||||
return self.decode_data(data=self.data, decoder=decoder)
|
|
||||||
|
|
||||||
def encode_object(self, object: Base):
|
def decode(self, decoder: Callable[[List[Any]], Any], **kwargs: Dict[str, Any]):
|
||||||
chunk = object.to_list()
|
return self.decode_data(data=self.data, decoder=decoder, **kwargs)
|
||||||
chunk.insert(0, len(chunk))
|
|
||||||
self.data.extend(chunk)
|
def encode_object(self, obj: Base):
|
||||||
|
encoded = obj.to_list()
|
||||||
|
encoded.insert(0, len(encoded))
|
||||||
|
self.data.extend(encoded)
|
||||||
|
|
||||||
|
|
||||||
class CurveArray(ObjectArray):
|
class CurveArray(ObjectArray):
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def from_curve(cls, curve: Base) -> 'CurveArray':
|
def from_curve(cls, curve: Base) -> "CurveArray":
|
||||||
crv_array = cls()
|
crv_array = cls()
|
||||||
crv_array.data = curve.to_list()
|
crv_array.data = curve.to_list()
|
||||||
return crv_array
|
return crv_array
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def from_curves(cls, curves: List[Base]) -> 'CurveArray':
|
def from_curves(cls, curves: List[Base]) -> "CurveArray":
|
||||||
data = []
|
data = []
|
||||||
for curve in curves:
|
for curve in curves:
|
||||||
curve_list = curve.to_list()
|
curve_list = curve.to_list()
|
||||||
@@ -119,8 +124,7 @@ class CurveArray(ObjectArray):
|
|||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def _curve_decoder(cls, data: List[float]) -> Base:
|
def _curve_decoder(cls, data: List[float]) -> Base:
|
||||||
crv_array = cls()
|
crv_array = cls(data)
|
||||||
crv_array.data = data
|
|
||||||
return crv_array.to_curve()
|
return crv_array.to_curve()
|
||||||
|
|
||||||
def to_curves(self) -> List[Base]:
|
def to_curves(self) -> List[Base]:
|
||||||
|
|||||||
@@ -1,5 +1,6 @@
|
|||||||
from specklepy.objects.geometry import Point
|
from enum import Enum
|
||||||
from typing import List
|
from typing import List
|
||||||
|
from specklepy.objects.geometry import Point
|
||||||
|
|
||||||
from .base import Base
|
from .base import Base
|
||||||
|
|
||||||
@@ -19,11 +20,19 @@ class FakeGeo(Base, chunkable={"dots": 50}, detachable={"pointslist"}):
|
|||||||
dots: List[int] = None
|
dots: List[int] = None
|
||||||
|
|
||||||
|
|
||||||
|
class FakeDirection(Enum):
|
||||||
|
NORTH = 1
|
||||||
|
EAST = 2
|
||||||
|
SOUTH = 3
|
||||||
|
WEST = 4
|
||||||
|
|
||||||
|
|
||||||
class FakeMesh(FakeGeo, chunkable=CHUNKABLE_PROPS, detachable=DETACHABLE):
|
class FakeMesh(FakeGeo, chunkable=CHUNKABLE_PROPS, detachable=DETACHABLE):
|
||||||
vertices: List[float] = None
|
vertices: List[float] = None
|
||||||
faces: List[int] = None
|
faces: List[int] = None
|
||||||
colors: List[int] = None
|
colors: List[int] = None
|
||||||
textureCoordinates: List[float] = None
|
textureCoordinates: List[float] = None
|
||||||
|
cardinal_dir: FakeDirection = None
|
||||||
test_bases: List[Base] = None
|
test_bases: List[Base] = None
|
||||||
detach_this: Base = None
|
detach_this: Base = None
|
||||||
detached_list: List[Base] = None
|
detached_list: List[Base] = None
|
||||||
|
|||||||
+326
-171
@@ -33,6 +33,7 @@ class Point(Base, speckle_type=GEOMETRY + "Point"):
|
|||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def from_list(cls, args: List[float]) -> "Point":
|
def from_list(cls, args: List[float]) -> "Point":
|
||||||
|
"""Create a new Point from a list of three floats representing the x, y, and z coordinates"""
|
||||||
return cls(x=args[0], y=args[1], z=args[2])
|
return cls(x=args[0], y=args[1], z=args[2])
|
||||||
|
|
||||||
def to_list(self) -> List[Any]:
|
def to_list(self) -> List[Any]:
|
||||||
@@ -40,6 +41,7 @@ class Point(Base, speckle_type=GEOMETRY + "Point"):
|
|||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def from_coords(cls, x: float = 0.0, y: float = 0.0, z: float = 0.0):
|
def from_coords(cls, x: float = 0.0, y: float = 0.0, z: float = 0.0):
|
||||||
|
"""Create a new Point from x, y, and z values"""
|
||||||
pt = Point()
|
pt = Point()
|
||||||
pt.x, pt.y, pt.z = x, y, z
|
pt.x, pt.y, pt.z = x, y, z
|
||||||
return pt
|
return pt
|
||||||
@@ -62,19 +64,21 @@ class Plane(Base, speckle_type=GEOMETRY + "Plane"):
|
|||||||
@classmethod
|
@classmethod
|
||||||
def from_list(cls, args: List[Any]) -> "Plane":
|
def from_list(cls, args: List[Any]) -> "Plane":
|
||||||
return cls(
|
return cls(
|
||||||
origin=Point.from_list(args[0:3]),
|
origin=Point.from_list(args[:3]),
|
||||||
normal=Vector.from_list(args[3:6]),
|
normal=Vector.from_list(args[3:6]),
|
||||||
xdir=Vector.from_list(args[6:9]),
|
xdir=Vector.from_list(args[6:9]),
|
||||||
ydir=Vector.from_list(args[9:12]),
|
ydir=Vector.from_list(args[9:12]),
|
||||||
|
units=get_units_from_encoding(args[-1]),
|
||||||
)
|
)
|
||||||
|
|
||||||
def to_list(self) -> List[Any]:
|
def to_list(self) -> List[Any]:
|
||||||
encoded = []
|
return [
|
||||||
encoded.extend(self.origin.to_list())
|
*self.origin.to_list(),
|
||||||
encoded.extend(self.normal.to_list())
|
*self.normal.to_list(),
|
||||||
encoded.extend(self.xdir.to_list())
|
*self.xdir.to_list(),
|
||||||
encoded.extend(self.ydir.to_list())
|
*self.ydir.to_list(),
|
||||||
return encoded
|
get_encoding_from_units(self.units),
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
class Box(Base, speckle_type=GEOMETRY + "Box"):
|
class Box(Base, speckle_type=GEOMETRY + "Box"):
|
||||||
@@ -96,17 +100,21 @@ class Line(Base, speckle_type=GEOMETRY + "Line"):
|
|||||||
@classmethod
|
@classmethod
|
||||||
def from_list(cls, args: List[Any]) -> "Line":
|
def from_list(cls, args: List[Any]) -> "Line":
|
||||||
return cls(
|
return cls(
|
||||||
start=Point.from_list(args[0:3]),
|
start=Point.from_list(args[1:4]),
|
||||||
end=Point.from_list(args[3:6]),
|
end=Point.from_list(args[4:7]),
|
||||||
domain=Interval.from_list(args[6:9]),
|
domain=Interval.from_list(args[7:10]),
|
||||||
|
units=get_units_from_encoding(args[-1]),
|
||||||
)
|
)
|
||||||
|
|
||||||
def to_list(self) -> List[Any]:
|
def to_list(self) -> List[Any]:
|
||||||
encoded = []
|
domain = self.domain.to_list() if self.domain else [0, 1]
|
||||||
encoded.extend(self.start.to_list())
|
return [
|
||||||
encoded.extend(self.end.to_list())
|
CurveTypeEncoding.Line.value,
|
||||||
encoded.extend(self.domain.to_list())
|
*self.start.to_list(),
|
||||||
return encoded
|
*self.end.to_list(),
|
||||||
|
*domain,
|
||||||
|
get_encoding_from_units(self.units),
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
class Arc(Base, speckle_type=GEOMETRY + "Arc"):
|
class Arc(Base, speckle_type=GEOMETRY + "Arc"):
|
||||||
@@ -132,20 +140,26 @@ class Arc(Base, speckle_type=GEOMETRY + "Arc"):
|
|||||||
angleRadians=args[4],
|
angleRadians=args[4],
|
||||||
domain=Interval.from_list(args[5:7]),
|
domain=Interval.from_list(args[5:7]),
|
||||||
plane=Plane.from_list(args[7:20]),
|
plane=Plane.from_list(args[7:20]),
|
||||||
|
startPoint=Point.from_list(args[20:23]),
|
||||||
|
midPoint=Point.from_list(args[23:26]),
|
||||||
|
endPoint=Point.from_list(args[26:29]),
|
||||||
units=get_units_from_encoding(args[-1]),
|
units=get_units_from_encoding(args[-1]),
|
||||||
)
|
)
|
||||||
|
|
||||||
def to_list(self) -> List[Any]:
|
def to_list(self) -> List[Any]:
|
||||||
encoded = []
|
return [
|
||||||
encoded.append(CurveTypeEncoding.Arc.value)
|
CurveTypeEncoding.Arc.value,
|
||||||
encoded.append(self.radius)
|
self.radius,
|
||||||
encoded.append(self.startAngle)
|
self.startAngle,
|
||||||
encoded.append(self.endAngle)
|
self.endAngle,
|
||||||
encoded.append(self.angleRadians)
|
self.angleRadians,
|
||||||
encoded.extend(self.domain.to_list())
|
*self.domain.to_list(),
|
||||||
encoded.extend(self.plane.to_list())
|
*self.plane.to_list(),
|
||||||
encoded.append(get_encoding_from_units(self.units))
|
*self.startPoint.to_list(),
|
||||||
return encoded
|
*self.midPoint.to_list(),
|
||||||
|
*self.endPoint.to_list(),
|
||||||
|
get_encoding_from_units(self.units),
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
class Circle(Base, speckle_type=GEOMETRY + "Circle"):
|
class Circle(Base, speckle_type=GEOMETRY + "Circle"):
|
||||||
@@ -166,13 +180,13 @@ class Circle(Base, speckle_type=GEOMETRY + "Circle"):
|
|||||||
)
|
)
|
||||||
|
|
||||||
def to_list(self) -> List[Any]:
|
def to_list(self) -> List[Any]:
|
||||||
encoded = []
|
return [
|
||||||
encoded.append(CurveTypeEncoding.Circle.value)
|
CurveTypeEncoding.Circle.value,
|
||||||
encoded.append(self.radius),
|
self.radius,
|
||||||
encoded.extend(self.domain.to_list())
|
*self.domain.to_list(),
|
||||||
encoded.extend(self.plane.to_list())
|
*self.plane.to_list(),
|
||||||
encoded.append(get_encoding_from_units(self.units))
|
get_encoding_from_units(self.units),
|
||||||
return encoded
|
]
|
||||||
|
|
||||||
|
|
||||||
class Ellipse(Base, speckle_type=GEOMETRY + "Ellipse"):
|
class Ellipse(Base, speckle_type=GEOMETRY + "Ellipse"):
|
||||||
@@ -196,14 +210,14 @@ class Ellipse(Base, speckle_type=GEOMETRY + "Ellipse"):
|
|||||||
)
|
)
|
||||||
|
|
||||||
def to_list(self) -> List[Any]:
|
def to_list(self) -> List[Any]:
|
||||||
encoded = []
|
return [
|
||||||
encoded.append(CurveTypeEncoding.Ellipse.value)
|
CurveTypeEncoding.Ellipse.value,
|
||||||
encoded.append(self.firstRadius)
|
self.firstRadius,
|
||||||
encoded.append(self.secondRadius)
|
self.secondRadius,
|
||||||
encoded.extend(self.domain.to_list())
|
*self.domain.to_list(),
|
||||||
encoded.extend(self.plane.to_list())
|
*self.plane.to_list(),
|
||||||
encoded.append(get_encoding_from_units(self.units))
|
get_encoding_from_units(self.units),
|
||||||
return encoded
|
]
|
||||||
|
|
||||||
|
|
||||||
class Polyline(Base, speckle_type=GEOMETRY + "Polyline", chunkable={"value": 20000}):
|
class Polyline(Base, speckle_type=GEOMETRY + "Polyline", chunkable={"value": 20000}):
|
||||||
@@ -216,6 +230,7 @@ class Polyline(Base, speckle_type=GEOMETRY + "Polyline", chunkable={"value": 200
|
|||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def from_points(cls, points: List[Point]):
|
def from_points(cls, points: List[Point]):
|
||||||
|
"""Create a new Polyline from a list of Points"""
|
||||||
polyline = cls()
|
polyline = cls()
|
||||||
polyline.units = points[0].units
|
polyline.units = points[0].units
|
||||||
polyline.value = []
|
polyline.value = []
|
||||||
@@ -234,14 +249,14 @@ class Polyline(Base, speckle_type=GEOMETRY + "Polyline", chunkable={"value": 200
|
|||||||
)
|
)
|
||||||
|
|
||||||
def to_list(self) -> List[Any]:
|
def to_list(self) -> List[Any]:
|
||||||
encoded = []
|
return [
|
||||||
encoded.append(CurveTypeEncoding.Polyline.value)
|
CurveTypeEncoding.Polyline.value,
|
||||||
encoded.append(int(self.closed))
|
int(self.closed),
|
||||||
encoded.extend(self.domain.to_list())
|
*self.domain.to_list(),
|
||||||
encoded.append(len(self.value))
|
len(self.value),
|
||||||
encoded.extend(self.value)
|
*self.value,
|
||||||
encoded.append(get_encoding_from_units(self.units))
|
get_encoding_from_units(self.units),
|
||||||
return encoded
|
]
|
||||||
|
|
||||||
def as_points(self) -> List[Point]:
|
def as_points(self) -> List[Point]:
|
||||||
"""Converts the `value` attribute to a list of Points"""
|
"""Converts the `value` attribute to a list of Points"""
|
||||||
@@ -290,9 +305,9 @@ class Curve(
|
|||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def from_list(cls, args: List[Any]) -> "Curve":
|
def from_list(cls, args: List[Any]) -> "Curve":
|
||||||
point_count = args[7]
|
point_count = int(args[7])
|
||||||
weights_count = args[8]
|
weights_count = int(args[8])
|
||||||
knots_count = args[9]
|
knots_count = int(args[9])
|
||||||
|
|
||||||
points_start = 10
|
points_start = 10
|
||||||
weights_start = 10 + point_count
|
weights_start = 10 + point_count
|
||||||
@@ -300,7 +315,7 @@ class Curve(
|
|||||||
knots_end = knots_start + knots_count
|
knots_end = knots_start + knots_count
|
||||||
|
|
||||||
return cls(
|
return cls(
|
||||||
degree=args[1],
|
degree=int(args[1]),
|
||||||
periodic=bool(args[2]),
|
periodic=bool(args[2]),
|
||||||
rational=bool(args[3]),
|
rational=bool(args[3]),
|
||||||
closed=bool(args[4]),
|
closed=bool(args[4]),
|
||||||
@@ -312,21 +327,21 @@ class Curve(
|
|||||||
)
|
)
|
||||||
|
|
||||||
def to_list(self) -> List[Any]:
|
def to_list(self) -> List[Any]:
|
||||||
encoded = []
|
return [
|
||||||
encoded.append(CurveTypeEncoding.Curve.value)
|
CurveTypeEncoding.Curve.value,
|
||||||
encoded.append(self.degree)
|
self.degree,
|
||||||
encoded.append(int(self.periodic))
|
int(self.periodic),
|
||||||
encoded.append(int(self.rational))
|
int(self.rational),
|
||||||
encoded.append(int(self.closed))
|
int(self.closed),
|
||||||
encoded.extend(self.domain.to_list())
|
*self.domain.to_list(),
|
||||||
encoded.append(len(self.points))
|
len(self.points),
|
||||||
encoded.append(len(self.weights))
|
len(self.weights),
|
||||||
encoded.append(len(self.knots))
|
len(self.knots),
|
||||||
encoded.extend(self.points)
|
*self.points,
|
||||||
encoded.extend(self.weights)
|
*self.weights,
|
||||||
encoded.extend(self.knots)
|
*self.knots,
|
||||||
encoded.append(get_encoding_from_units(self.units))
|
get_encoding_from_units(self.units),
|
||||||
return encoded
|
]
|
||||||
|
|
||||||
|
|
||||||
class Polycurve(Base, speckle_type=GEOMETRY + "Polycurve"):
|
class Polycurve(Base, speckle_type=GEOMETRY + "Polycurve"):
|
||||||
@@ -339,8 +354,7 @@ class Polycurve(Base, speckle_type=GEOMETRY + "Polycurve"):
|
|||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def from_list(cls, args: List[Any]) -> "Polycurve":
|
def from_list(cls, args: List[Any]) -> "Polycurve":
|
||||||
curve_arrays = CurveArray()
|
curve_arrays = CurveArray(args[5:-1])
|
||||||
curve_arrays.data = args[4:-1]
|
|
||||||
return cls(
|
return cls(
|
||||||
closed=bool(args[1]),
|
closed=bool(args[1]),
|
||||||
domain=Interval.from_list(args[2:4]),
|
domain=Interval.from_list(args[2:4]),
|
||||||
@@ -349,14 +363,15 @@ class Polycurve(Base, speckle_type=GEOMETRY + "Polycurve"):
|
|||||||
)
|
)
|
||||||
|
|
||||||
def to_list(self) -> List[Any]:
|
def to_list(self) -> List[Any]:
|
||||||
encoded = []
|
curve_array = CurveArray.from_curves(self.segments).data
|
||||||
encoded.append(CurveTypeEncoding.Polycurve.value)
|
return [
|
||||||
encoded.append(int(self.closed))
|
CurveTypeEncoding.Polycurve.value,
|
||||||
encoded.extend(self.domain.to_list())
|
int(self.closed),
|
||||||
curve_array = CurveArray.from_curves(self.segments)
|
*self.domain.to_list(),
|
||||||
encoded.extend(curve_array.data)
|
len(curve_array),
|
||||||
encoded.append(get_encoding_from_units(self.units))
|
*curve_array,
|
||||||
return encoded
|
get_encoding_from_units(self.units),
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
class Extrusion(Base, speckle_type=GEOMETRY + "Extrusion"):
|
class Extrusion(Base, speckle_type=GEOMETRY + "Extrusion"):
|
||||||
@@ -391,6 +406,28 @@ class Mesh(
|
|||||||
area: float = None
|
area: float = None
|
||||||
volume: float = None
|
volume: float = None
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def create(
|
||||||
|
cls,
|
||||||
|
vertices: List[float],
|
||||||
|
faces: List[int],
|
||||||
|
colors: List[int] = None,
|
||||||
|
texture_coordinates: List[float] = None,
|
||||||
|
) -> "Mesh":
|
||||||
|
"""
|
||||||
|
Create a new Mesh from lists representing its vertices, faces,
|
||||||
|
colors (optional), and texture coordinates (optional).
|
||||||
|
|
||||||
|
This will initialise empty lists for colors and texture coordinates
|
||||||
|
if you do not provide any.
|
||||||
|
"""
|
||||||
|
return cls(
|
||||||
|
vertices=vertices,
|
||||||
|
faces=faces,
|
||||||
|
colors=colors or [],
|
||||||
|
textureCoordinates=texture_coordinates or [],
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
class Surface(Base, speckle_type=GEOMETRY + "Surface"):
|
class Surface(Base, speckle_type=GEOMETRY + "Surface"):
|
||||||
degreeU: int = None
|
degreeU: int = None
|
||||||
@@ -435,46 +472,65 @@ class Surface(Base, speckle_type=GEOMETRY + "Surface"):
|
|||||||
)
|
)
|
||||||
|
|
||||||
def to_list(self) -> List[Any]:
|
def to_list(self) -> List[Any]:
|
||||||
encoded = []
|
return [
|
||||||
encoded.append(self.degreeU)
|
self.degreeU,
|
||||||
encoded.append(self.degreeV)
|
self.degreeV,
|
||||||
encoded.append(self.countU)
|
self.countU,
|
||||||
encoded.append(self.countV)
|
self.countV,
|
||||||
encoded.append(int(self.rational))
|
int(self.rational),
|
||||||
encoded.append(int(self.closedU))
|
int(self.closedU),
|
||||||
encoded.append(int(self.closedV))
|
int(self.closedV),
|
||||||
encoded.extend(self.domainU.to_list())
|
*self.domainU.to_list(),
|
||||||
encoded.extend(self.domainV.to_list())
|
*self.domainV.to_list(),
|
||||||
encoded.append(len(self.pointData))
|
len(self.pointData),
|
||||||
encoded.append(len(self.knotsU))
|
len(self.knotsU),
|
||||||
encoded.append(len(self.knotsV))
|
len(self.knotsV),
|
||||||
encoded.extend(self.pointData)
|
*self.pointData,
|
||||||
encoded.extend(self.knotsU)
|
*self.knotsU,
|
||||||
encoded.extend(self.knotsV)
|
*self.knotsV,
|
||||||
encoded.append(get_encoding_from_units(self.units))
|
get_encoding_from_units(self.units),
|
||||||
return encoded
|
]
|
||||||
|
|
||||||
|
|
||||||
class BrepFace(Base, speckle_type=GEOMETRY + "BrepFace"):
|
class BrepFace(Base, speckle_type=GEOMETRY + "BrepFace"):
|
||||||
_Brep: "Brep" = None
|
_Brep: "Brep" = None
|
||||||
SurfaceIndex: int = None
|
SurfaceIndex: int = None
|
||||||
LoopIndices: List[int] = None
|
|
||||||
OuterLoopIndex: int = None
|
OuterLoopIndex: int = None
|
||||||
OrientationReversed: bool = None
|
OrientationReversed: bool = None
|
||||||
|
LoopIndices: List[int] = None
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def _outer_loop(self):
|
def _outer_loop(self):
|
||||||
return self._Brep.Loops[self.OuterLoopIndex]
|
return self._Brep.Loops[self.OuterLoopIndex] # pylint: disable=no-member
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def _surface(self):
|
def _surface(self):
|
||||||
return self._Brep.Surfaces[self.SurfaceIndex]
|
return self._Brep.Surfaces[self.SurfaceIndex] # pylint: disable=no-member
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def _loops(self):
|
def _loops(self):
|
||||||
if self.LoopIndices:
|
if self.LoopIndices:
|
||||||
|
# pylint: disable=not-an-iterable, no-member
|
||||||
return [self._Brep.Loops[i] for i in self.LoopIndices]
|
return [self._Brep.Loops[i] for i in self.LoopIndices]
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def from_list(cls, args: List[Any], brep: "Brep" = None) -> "BrepFace":
|
||||||
|
return cls(
|
||||||
|
_Brep=brep,
|
||||||
|
SurfaceIndex=args[0],
|
||||||
|
OuterLoopIndex=args[1],
|
||||||
|
OrientationReversed=bool(args[2]),
|
||||||
|
LoopIndices=args[3:],
|
||||||
|
)
|
||||||
|
|
||||||
|
def to_list(self) -> List[Any]:
|
||||||
|
return [
|
||||||
|
self.SurfaceIndex,
|
||||||
|
self.OuterLoopIndex,
|
||||||
|
int(self.OrientationReversed),
|
||||||
|
*self.LoopIndices,
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
class BrepEdge(Base, speckle_type=GEOMETRY + "BrepEdge"):
|
class BrepEdge(Base, speckle_type=GEOMETRY + "BrepEdge"):
|
||||||
_Brep: "Brep" = None
|
_Brep: "Brep" = None
|
||||||
@@ -496,18 +552,59 @@ class BrepEdge(Base, speckle_type=GEOMETRY + "BrepEdge"):
|
|||||||
@property
|
@property
|
||||||
def _trims(self):
|
def _trims(self):
|
||||||
if self.TrimIndices:
|
if self.TrimIndices:
|
||||||
|
# pylint: disable=not-an-iterable
|
||||||
return [self._Brep.Trims[i] for i in self.TrimIndices]
|
return [self._Brep.Trims[i] for i in self.TrimIndices]
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def _curve(self):
|
def _curve(self):
|
||||||
return self._Brep.Curve3D[self.Curve3dIndex]
|
return self._Brep.Curve3D[self.Curve3dIndex]
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def from_list(cls, args: List[Any], brep: "Brep" = None) -> "BrepEdge":
|
||||||
|
domain_start = args[4]
|
||||||
|
domain_end = args[5]
|
||||||
|
domain = (
|
||||||
|
Interval(start=domain_start, end=domain_end)
|
||||||
|
if None not in (domain_start, domain_end)
|
||||||
|
else None
|
||||||
|
)
|
||||||
|
return cls(
|
||||||
|
_Brep=brep,
|
||||||
|
Curve3dIndex=int(args[0]),
|
||||||
|
TrimIndices=[int(t) for t in args[6:]],
|
||||||
|
StartIndex=int(args[1]),
|
||||||
|
EndIndex=int(args[2]),
|
||||||
|
ProxyCurveIsReversed=bool(args[3]),
|
||||||
|
Domain=domain,
|
||||||
|
)
|
||||||
|
|
||||||
|
def to_list(self) -> List[Any]:
|
||||||
|
return [
|
||||||
|
self.Curve3dIndex,
|
||||||
|
self.StartIndex,
|
||||||
|
self.EndIndex,
|
||||||
|
int(self.ProxyCurveIsReversed),
|
||||||
|
self.Domain.start,
|
||||||
|
self.Domain.end,
|
||||||
|
*self.TrimIndices,
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
class BrepLoopType(int, Enum):
|
||||||
|
Unknown = 0
|
||||||
|
Outer = 1
|
||||||
|
Inner = 2
|
||||||
|
Slit = 3
|
||||||
|
CurveOnSurface = 4
|
||||||
|
PointOnSurface = 5
|
||||||
|
|
||||||
|
|
||||||
class BrepLoop(Base, speckle_type=GEOMETRY + "BrepLoop"):
|
class BrepLoop(Base, speckle_type=GEOMETRY + "BrepLoop"):
|
||||||
_Brep: "Brep" = None
|
_Brep: "Brep" = None
|
||||||
FaceIndex: int = None
|
FaceIndex: int = None
|
||||||
TrimIndices: List[int] = None
|
TrimIndices: List[int] = None
|
||||||
Type: str = None
|
Type: BrepLoopType = None
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def _face(self):
|
def _face(self):
|
||||||
@@ -516,10 +613,27 @@ class BrepLoop(Base, speckle_type=GEOMETRY + "BrepLoop"):
|
|||||||
@property
|
@property
|
||||||
def _trims(self):
|
def _trims(self):
|
||||||
if self.TrimIndices:
|
if self.TrimIndices:
|
||||||
|
# pylint: disable=not-an-iterable
|
||||||
return [self._Brep.Trims[i] for i in self.TrimIndices]
|
return [self._Brep.Trims[i] for i in self.TrimIndices]
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def from_list(cls, args: List[any], brep: "Brep" = None):
|
||||||
|
return cls(
|
||||||
|
_Brep=brep,
|
||||||
|
FaceIndex=args[0],
|
||||||
|
Type=BrepLoopType(args[1]),
|
||||||
|
TrimIndices=args[2:],
|
||||||
|
)
|
||||||
|
|
||||||
class BrepTrimTypeEnum(int, Enum):
|
def to_list(self) -> List[int]:
|
||||||
|
return [
|
||||||
|
self.FaceIndex,
|
||||||
|
self.Type.value,
|
||||||
|
*self.TrimIndices,
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
|
class BrepTrimType(int, Enum):
|
||||||
Unknown = 0
|
Unknown = 0
|
||||||
Boundary = 1
|
Boundary = 1
|
||||||
Mated = 2
|
Mated = 2
|
||||||
@@ -539,29 +653,35 @@ class BrepTrim(Base, speckle_type=GEOMETRY + "BrepTrim"):
|
|||||||
LoopIndex: int = None
|
LoopIndex: int = None
|
||||||
CurveIndex: int = None
|
CurveIndex: int = None
|
||||||
IsoStatus: int = None
|
IsoStatus: int = None
|
||||||
TrimType: str = None
|
TrimType: BrepTrimType = None
|
||||||
IsReversed: bool = None
|
IsReversed: bool = None
|
||||||
Domain: Interval = None
|
Domain: Interval = None
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def _face(self):
|
def _face(self):
|
||||||
return self._Brep.Faces[self.FaceIndex]
|
if self._Brep:
|
||||||
|
return self._Brep.Faces[self.FaceIndex] # pylint: disable=no-member
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def _loop(self):
|
def _loop(self):
|
||||||
return self._Brep.Loops[self.LoopIndex]
|
if self._Brep:
|
||||||
|
return self._Brep.Loops[self.LoopIndex] # pylint: disable=no-member
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def _edge(self):
|
def _edge(self):
|
||||||
return self._Brep.Edges[self.EdgeIndex] if self.EdgeIndex != -1 else None
|
if self._Brep:
|
||||||
|
# pylint: disable=no-member
|
||||||
|
return self._Brep.Edges[self.EdgeIndex] if self.EdgeIndex != -1 else None
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def _curve_2d(self):
|
def _curve_2d(self):
|
||||||
return self._Brep.Curve2D[self.CurveIndex]
|
if self._Brep:
|
||||||
|
return self._Brep.Curve2D[self.CurveIndex] # pylint: disable=no-member
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def from_list(cls, args: List[Any]) -> "BrepTrim":
|
def from_list(cls, args: List[Any], brep: "Brep" = None) -> "BrepTrim":
|
||||||
return cls(
|
return cls(
|
||||||
|
_Brep=brep,
|
||||||
EdgeIndex=args[0],
|
EdgeIndex=args[0],
|
||||||
StartIndex=args[1],
|
StartIndex=args[1],
|
||||||
EndIndex=args[2],
|
EndIndex=args[2],
|
||||||
@@ -569,49 +689,62 @@ class BrepTrim(Base, speckle_type=GEOMETRY + "BrepTrim"):
|
|||||||
LoopIndex=args[4],
|
LoopIndex=args[4],
|
||||||
CurveIndex=args[5],
|
CurveIndex=args[5],
|
||||||
IsoStatus=args[6],
|
IsoStatus=args[6],
|
||||||
TrimType=BrepTrimTypeEnum(args[7]).name,
|
TrimType=BrepTrimType(args[7]),
|
||||||
IsReversed=bool(args[8]),
|
IsReversed=bool(args[8]),
|
||||||
)
|
)
|
||||||
|
|
||||||
def to_list(self) -> List[Any]:
|
def to_list(self) -> List[Any]:
|
||||||
encoded = []
|
return [
|
||||||
encoded.append(self.EdgeIndex)
|
self.EdgeIndex,
|
||||||
encoded.append(self.StartIndex)
|
self.StartIndex,
|
||||||
encoded.append(self.EndIndex)
|
self.EndIndex,
|
||||||
encoded.append(self.FaceIndex)
|
self.FaceIndex,
|
||||||
encoded.append(self.LoopIndex)
|
self.LoopIndex,
|
||||||
encoded.append(self.CurveIndex)
|
self.CurveIndex,
|
||||||
encoded.append(self.IsoStatus)
|
self.IsoStatus,
|
||||||
encoded.append(getattr(BrepTrimTypeEnum, self.TrimType).value)
|
self.TrimType.value,
|
||||||
encoded.append(self.IsReversed)
|
int(self.IsReversed),
|
||||||
return encoded
|
]
|
||||||
|
|
||||||
|
|
||||||
class Brep(
|
class Brep(
|
||||||
Base,
|
Base,
|
||||||
speckle_type=GEOMETRY + "Brep",
|
speckle_type=GEOMETRY + "Brep",
|
||||||
chunkable={
|
chunkable={
|
||||||
"SurfacesValue": 200,
|
"SurfacesValue": 31250,
|
||||||
"Curve3DValues": 200,
|
"Curve3DValues": 31250,
|
||||||
"Curve2DValues": 200,
|
"Curve2DValues": 31250,
|
||||||
"VerticesValue": 5000,
|
"VerticesValue": 31250,
|
||||||
"Edges": 5000,
|
"EdgesValue": 62500,
|
||||||
"Loops": 5000,
|
"LoopsValue": 62500,
|
||||||
"TrimsValue": 5000,
|
"FacesValue": 62500,
|
||||||
"Faces": 5000,
|
"TrimsValue": 62500,
|
||||||
},
|
},
|
||||||
detachable={"displayValue"},
|
detachable={"displayValue"},
|
||||||
serialize_ignore={"Surfaces", "Curve3D", "Curve2D", "Vertices", "Trims"},
|
serialize_ignore={
|
||||||
|
"Surfaces",
|
||||||
|
"Curve3D",
|
||||||
|
"Curve2D",
|
||||||
|
"Vertices",
|
||||||
|
"Trims",
|
||||||
|
"Edges",
|
||||||
|
"Loops",
|
||||||
|
"Faces",
|
||||||
|
},
|
||||||
):
|
):
|
||||||
provenance: str = None
|
provenance: str = None
|
||||||
bbox: Box = None
|
bbox: Box = None
|
||||||
area: float = None
|
area: float = None
|
||||||
volume: float = None
|
volume: float = None
|
||||||
displayValue: Mesh = None
|
_displayValue: List[Mesh] = None
|
||||||
Surfaces: List[Surface] = None
|
Surfaces: List[Surface] = None
|
||||||
Curve3D: List[Base] = None
|
Curve3D: List[Base] = None
|
||||||
Curve2D: List[Base] = None
|
Curve2D: List[Base] = None
|
||||||
Vertices: List[Point] = None
|
Vertices: List[Point] = None
|
||||||
|
Edges: List[BrepEdge] = None
|
||||||
|
Loops: List[BrepLoop] = None
|
||||||
|
Faces: List[BrepFace] = None
|
||||||
|
Trims: List[BrepTrim] = None
|
||||||
IsClosed: bool = None
|
IsClosed: bool = None
|
||||||
Orientation: int = None
|
Orientation: int = None
|
||||||
|
|
||||||
@@ -620,65 +753,89 @@ class Brep(
|
|||||||
return children
|
return children
|
||||||
|
|
||||||
for child in children:
|
for child in children:
|
||||||
child._Brep = self
|
child._Brep = self # pylint: disable=protected-access
|
||||||
return children
|
return children
|
||||||
|
|
||||||
|
# set as prop for now for backwards compatibility
|
||||||
@property
|
@property
|
||||||
def Edges(self) -> List[BrepEdge]:
|
def displayValue(self) -> List[Mesh]:
|
||||||
return self._inject_self_into_children(self._Edges)
|
return self._displayValue
|
||||||
|
|
||||||
@Edges.setter
|
@displayValue.setter
|
||||||
def Edges(self, value: List[BrepEdge]):
|
def displayValue(self, value):
|
||||||
self._Edges = value
|
if isinstance(value, Mesh):
|
||||||
|
self._displayValue = [value]
|
||||||
|
elif isinstance(value, list):
|
||||||
|
self._displayValue = value
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def Loops(self) -> List[BrepLoop]:
|
def EdgesValue(self) -> List[BrepEdge]:
|
||||||
return self._inject_self_into_children(self._Loops)
|
return None if self.Edges is None else ObjectArray.from_objects(self.Edges).data
|
||||||
|
|
||||||
@Loops.setter
|
@EdgesValue.setter
|
||||||
def Loops(self, value: List[BrepLoop]):
|
def EdgesValue(self, value: List[float]):
|
||||||
self._Loops = value
|
if not value:
|
||||||
|
return
|
||||||
|
|
||||||
|
self.Edges = ObjectArray.decode_data(value, BrepEdge.from_list, brep=self)
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def Faces(self) -> List[BrepFace]:
|
def LoopsValue(self) -> List[BrepLoop]:
|
||||||
return self._inject_self_into_children(self._Faces)
|
return None if self.Loops is None else ObjectArray.from_objects(self.Loops).data
|
||||||
|
|
||||||
@Faces.setter
|
@LoopsValue.setter
|
||||||
def Faces(self, value: List[BrepFace]):
|
def LoopsValue(self, value: List[int]):
|
||||||
self._Faces = value
|
if not value:
|
||||||
|
return
|
||||||
|
|
||||||
|
self.Loops = ObjectArray.decode_data(value, BrepLoop.from_list, brep=self)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def FacesValue(self) -> List[int]:
|
||||||
|
return None if self.Faces is None else ObjectArray.from_objects(self.Faces).data
|
||||||
|
|
||||||
|
@FacesValue.setter
|
||||||
|
def FacesValue(self, value: List[int]):
|
||||||
|
if not value:
|
||||||
|
return
|
||||||
|
|
||||||
|
self.Faces = ObjectArray.decode_data(value, BrepFace.from_list, brep=self)
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def SurfacesValue(self) -> List[float]:
|
def SurfacesValue(self) -> List[float]:
|
||||||
if self.Surfaces is None:
|
return (
|
||||||
return None
|
None
|
||||||
return ObjectArray.from_objects(self.Surfaces).data
|
if self.Surfaces is None
|
||||||
|
else ObjectArray.from_objects(self.Surfaces).data
|
||||||
|
)
|
||||||
|
|
||||||
@SurfacesValue.setter
|
@SurfacesValue.setter
|
||||||
def SurfacesValue(self, value: List[float]):
|
def SurfacesValue(self, value: List[float]):
|
||||||
|
if not value:
|
||||||
|
return
|
||||||
|
|
||||||
self.Surfaces = ObjectArray.decode_data(value, Surface.from_list)
|
self.Surfaces = ObjectArray.decode_data(value, Surface.from_list)
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def Curve3DValues(self) -> List[float]:
|
def Curve3DValues(self) -> List[float]:
|
||||||
if self.Curve3D is None:
|
return (
|
||||||
return None
|
None if self.Curve3D is None else CurveArray.from_curves(self.Curve3D).data
|
||||||
return CurveArray.from_curves(self.Curve3D).data
|
)
|
||||||
|
|
||||||
@Curve3DValues.setter
|
@Curve3DValues.setter
|
||||||
def Curve3DValues(self, value: List[float]):
|
def Curve3DValues(self, value: List[float]):
|
||||||
crv_array = CurveArray()
|
crv_array = CurveArray(value)
|
||||||
crv_array.data = value
|
|
||||||
self.Curve3D = crv_array.to_curves()
|
self.Curve3D = crv_array.to_curves()
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def Curve2DValues(self) -> List[Base]:
|
def Curve2DValues(self) -> List[Base]:
|
||||||
if self.Curve2D is None:
|
return (
|
||||||
return None
|
None if self.Curve2D is None else CurveArray.from_curves(self.Curve2D).data
|
||||||
return CurveArray.from_curves(self.Curve2D).data
|
)
|
||||||
|
|
||||||
@Curve2DValues.setter
|
@Curve2DValues.setter
|
||||||
def Curve2DValues(self, value: List[float]):
|
def Curve2DValues(self, value: List[float]):
|
||||||
crv_array = CurveArray()
|
crv_array = CurveArray(value)
|
||||||
crv_array.data = value
|
|
||||||
self.Curve2D = crv_array.to_curves()
|
self.Curve2D = crv_array.to_curves()
|
||||||
|
|
||||||
@property
|
@property
|
||||||
@@ -705,27 +862,25 @@ class Brep(
|
|||||||
|
|
||||||
self.Vertices = vertices
|
self.Vertices = vertices
|
||||||
|
|
||||||
@property
|
# TODO: can this be consistent with loops, edges, faces, curves, etc and prepend with the chunk list? needs to happen in sharp first
|
||||||
def Trims(self) -> List[BrepTrim]:
|
|
||||||
return self._inject_self_into_children(self._Trims)
|
|
||||||
|
|
||||||
@Trims.setter
|
|
||||||
def Trims(self, value: List[BrepTrim]):
|
|
||||||
self._Trims = value
|
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def TrimsValue(self) -> List[float]:
|
def TrimsValue(self) -> List[float]:
|
||||||
if self.Trims is None:
|
# return None if self.Trims is None else ObjectArray.from_objects(self.Trims).data
|
||||||
return None
|
if not self.Trims:
|
||||||
values = []
|
return
|
||||||
|
value = []
|
||||||
for trim in self.Trims:
|
for trim in self.Trims:
|
||||||
values.extend(trim.to_list())
|
value.extend(trim.to_list())
|
||||||
return values
|
return value
|
||||||
|
|
||||||
@TrimsValue.setter
|
@TrimsValue.setter
|
||||||
def TrimsValue(self, value: List[float]):
|
def TrimsValue(self, value: List[float]):
|
||||||
|
if not value:
|
||||||
|
return
|
||||||
|
|
||||||
|
# self.Trims = ObjectArray.decode_data(value, BrepTrim.from_list, brep=self)
|
||||||
self.Trims = [
|
self.Trims = [
|
||||||
BrepTrim.from_list(value[i : i + 9]) for i in range(0, len(value), 9)
|
BrepTrim.from_list(value[i : i + 9], self) for i in range(0, len(value), 9)
|
||||||
]
|
]
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -1,7 +1,28 @@
|
|||||||
|
from typing import Any, List
|
||||||
|
from specklepy.objects.geometry import Point, Vector
|
||||||
from .base import Base
|
from .base import Base
|
||||||
|
|
||||||
OTHER = "Objects.Other."
|
OTHER = "Objects.Other."
|
||||||
|
|
||||||
|
IDENTITY_TRANSFORM = [
|
||||||
|
1.0,
|
||||||
|
0.0,
|
||||||
|
0.0,
|
||||||
|
0.0,
|
||||||
|
0.0,
|
||||||
|
1.0,
|
||||||
|
0.0,
|
||||||
|
0.0,
|
||||||
|
0.0,
|
||||||
|
0.0,
|
||||||
|
1.0,
|
||||||
|
0.0,
|
||||||
|
0.0,
|
||||||
|
0.0,
|
||||||
|
0.0,
|
||||||
|
1.0,
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
class RenderMaterial(Base, speckle_type=OTHER + "RenderMaterial"):
|
class RenderMaterial(Base, speckle_type=OTHER + "RenderMaterial"):
|
||||||
name: str = None
|
name: str = None
|
||||||
@@ -10,3 +31,184 @@ class RenderMaterial(Base, speckle_type=OTHER + "RenderMaterial"):
|
|||||||
roughness: float = 1
|
roughness: float = 1
|
||||||
diffuse: int = -2894893 # light gray arbg
|
diffuse: int = -2894893 # light gray arbg
|
||||||
emissive: int = -16777216 # black arbg
|
emissive: int = -16777216 # black arbg
|
||||||
|
|
||||||
|
|
||||||
|
class Transform(
|
||||||
|
Base,
|
||||||
|
speckle_type=OTHER + "Transform",
|
||||||
|
serialize_ignore={"translation", "scaling", "is_identity"},
|
||||||
|
):
|
||||||
|
"""The 4x4 transformation matrix
|
||||||
|
|
||||||
|
The 3x3 sub-matrix determines scaling.
|
||||||
|
The 4th column defines translation, where the last value is a divisor (usually equal to 1).
|
||||||
|
"""
|
||||||
|
|
||||||
|
_value: List[float] = None
|
||||||
|
|
||||||
|
@property
|
||||||
|
def value(self) -> List[float]:
|
||||||
|
"""The transform matrix represented as a flat list of 16 floats"""
|
||||||
|
return self._value
|
||||||
|
|
||||||
|
@value.setter
|
||||||
|
def value(self, value: List[float]) -> None:
|
||||||
|
try:
|
||||||
|
value = [float(x) for x in value]
|
||||||
|
except (ValueError, TypeError) as error:
|
||||||
|
raise ValueError(
|
||||||
|
f"Could not create a Transform object with the requested value. Input must be a 16 element list of numbers. Value provided: {value}"
|
||||||
|
) from error
|
||||||
|
|
||||||
|
if len(value) != 16:
|
||||||
|
raise ValueError(
|
||||||
|
f"Could not create a Transform object: input list should be 16 floats long, but was {len(value)} long"
|
||||||
|
)
|
||||||
|
|
||||||
|
self._value = value
|
||||||
|
|
||||||
|
@property
|
||||||
|
def translation(self) -> List[float]:
|
||||||
|
"""The final column of the matrix which defines the translation"""
|
||||||
|
return [self._value[i] for i in (3, 7, 11, 15)]
|
||||||
|
|
||||||
|
@property
|
||||||
|
def scaling(self) -> List[float]:
|
||||||
|
"""The 3x3 scaling sub-matrix"""
|
||||||
|
return [self._value[i] for i in (0, 1, 2, 4, 5, 6, 8, 9, 10)]
|
||||||
|
|
||||||
|
@property
|
||||||
|
def is_identity(self) -> bool:
|
||||||
|
return self.value == IDENTITY_TRANSFORM
|
||||||
|
|
||||||
|
def apply_to_point(self, point: Point) -> Point:
|
||||||
|
"""Transform a single speckle Point
|
||||||
|
|
||||||
|
Arguments:
|
||||||
|
point {Point} -- the speckle Point to transform
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Point -- a new transformed point
|
||||||
|
"""
|
||||||
|
coords = self.apply_to_point_value([point.x, point.y, point.z])
|
||||||
|
return Point(x=coords[0], y=coords[1], z=coords[2], units=point.units)
|
||||||
|
|
||||||
|
def apply_to_point_value(self, point_value: List[float]) -> List[float]:
|
||||||
|
"""Transform a list of three floats representing a point
|
||||||
|
|
||||||
|
Arguments:
|
||||||
|
point_value {List[float]} -- a list of 3 floats
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
List[float] -- the list with the transform applied
|
||||||
|
"""
|
||||||
|
transformed = [
|
||||||
|
point_value[0] * self._value[i]
|
||||||
|
+ point_value[1] * self._value[i + 1]
|
||||||
|
+ point_value[2] * self._value[i + 2]
|
||||||
|
+ self._value[i + 3]
|
||||||
|
for i in range(0, 15, 4)
|
||||||
|
]
|
||||||
|
|
||||||
|
return [transformed[i] / transformed[3] for i in range(3)]
|
||||||
|
|
||||||
|
def apply_to_points(self, points: List[Point]) -> List[Point]:
|
||||||
|
"""Transform a list of speckle Points
|
||||||
|
|
||||||
|
Arguments:
|
||||||
|
points {List[Point]} -- the list of speckle Points to transform
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
List[Point] -- a new list of transformed points
|
||||||
|
"""
|
||||||
|
return [self.apply_to_point(point) for point in points]
|
||||||
|
|
||||||
|
def apply_to_points_values(self, points_value: List[float]) -> List[float]:
|
||||||
|
"""Transform a list of speckle Points
|
||||||
|
|
||||||
|
Arguments:
|
||||||
|
points {List[float]} -- a flat list of floats representing points to transform
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
List[float] -- a new transformed list
|
||||||
|
"""
|
||||||
|
if len(points_value) % 3 != 0:
|
||||||
|
raise ValueError(
|
||||||
|
"Cannot apply transform as the points list is malformed: expected length to be multiple of 3"
|
||||||
|
)
|
||||||
|
transformed = []
|
||||||
|
for i in range(0, len(points_value), 3):
|
||||||
|
transformed.extend(self.apply_to_point_value(points_value[i : i + 3]))
|
||||||
|
|
||||||
|
return transformed
|
||||||
|
|
||||||
|
def apply_to_vector(self, vector: Vector) -> Vector:
|
||||||
|
"""Transform a single speckle Vector
|
||||||
|
|
||||||
|
Arguments:
|
||||||
|
point {Vector} -- the speckle Vector to transform
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Vector -- a new transformed point
|
||||||
|
"""
|
||||||
|
coords = self.apply_to_vector_value([vector.x, vector.y, vector.z])
|
||||||
|
return Vector(x=coords[0], y=coords[1], z=coords[2], units=vector.units)
|
||||||
|
|
||||||
|
def apply_to_vector_value(self, vector_value: List[float]) -> List[float]:
|
||||||
|
"""Transform a list of three floats representing a vector
|
||||||
|
|
||||||
|
Arguments:
|
||||||
|
vector_value {List[float]} -- a list of 3 floats
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
List[float] -- the list with the transform applied
|
||||||
|
"""
|
||||||
|
return [
|
||||||
|
vector_value[0] * self._value[i]
|
||||||
|
+ vector_value[1] * self._value[i + 1]
|
||||||
|
+ vector_value[2] * self._value[i + 2]
|
||||||
|
for i in range(0, 15, 4)
|
||||||
|
][:3]
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def from_list(cls, value: List[float] = None) -> "Transform":
|
||||||
|
"""Returns a Transform object from a list of 16 numbers. If no value is provided, an identity transform will be returned.
|
||||||
|
|
||||||
|
Arguments:
|
||||||
|
value {List[float]} -- the matrix as a flat list of 16 numbers (defaults to the identity transform)
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Transform -- a complete transform object
|
||||||
|
"""
|
||||||
|
if not value:
|
||||||
|
value = IDENTITY_TRANSFORM
|
||||||
|
return cls(value=value)
|
||||||
|
|
||||||
|
|
||||||
|
class BlockDefinition(
|
||||||
|
Base, speckle_type=OTHER + "BlockDefinition", detachable={"geometry"}
|
||||||
|
):
|
||||||
|
name: str = None
|
||||||
|
basePoint: Point = None
|
||||||
|
geometry: List[Base] = None
|
||||||
|
|
||||||
|
|
||||||
|
class BlockInstance(
|
||||||
|
Base, speckle_type=OTHER + "BlockInstance", detachable={"blockDefinition"}
|
||||||
|
):
|
||||||
|
blockDefinition: BlockDefinition = None
|
||||||
|
transform: Transform = None
|
||||||
|
|
||||||
|
|
||||||
|
# TODO: prob move this into a built elements module, but just trialling this for now
|
||||||
|
class RevitParameter(Base, speckle_type="Objects.BuiltElements.Revit.Parameter"):
|
||||||
|
name: str = None
|
||||||
|
value: Any = None
|
||||||
|
applicationUnitType: str = None # eg UnitType UT_Length
|
||||||
|
applicationUnit: str = None # DisplayUnitType eg DUT_MILLIMITERS
|
||||||
|
applicationInternalName: str = (
|
||||||
|
None # BuiltInParameterName or GUID for shared parameter
|
||||||
|
)
|
||||||
|
isShared: bool = False
|
||||||
|
isReadOnly: bool = False
|
||||||
|
isTypeParameter: bool = False
|
||||||
|
|||||||
@@ -0,0 +1,40 @@
|
|||||||
|
"""Builtin Speckle object kit."""
|
||||||
|
|
||||||
|
from specklepy.objects.structural.analysis import *
|
||||||
|
from specklepy.objects.structural.properties import *
|
||||||
|
from specklepy.objects.structural.material import *
|
||||||
|
from specklepy.objects.structural.geometry import *
|
||||||
|
from specklepy.objects.structural.loading import *
|
||||||
|
from specklepy.objects.structural.axis import Axis
|
||||||
|
|
||||||
|
__all__ = [
|
||||||
|
"Element1D",
|
||||||
|
"Element2D",
|
||||||
|
"Element3D",
|
||||||
|
"Axis",
|
||||||
|
"Node",
|
||||||
|
"Restraint",
|
||||||
|
"Load",
|
||||||
|
"LoadBeam",
|
||||||
|
"LoadCase",
|
||||||
|
"LoadCombinations",
|
||||||
|
"LoadFace",
|
||||||
|
"LoadGravity",
|
||||||
|
"LoadNode",
|
||||||
|
"Model",
|
||||||
|
"ModelInfo",
|
||||||
|
"ModelSettings",
|
||||||
|
"ModelUnits",
|
||||||
|
"Concrete",
|
||||||
|
"Material",
|
||||||
|
"Steel",
|
||||||
|
"Timber",
|
||||||
|
"Property",
|
||||||
|
"Property1D",
|
||||||
|
"Property2D",
|
||||||
|
"Property3D",
|
||||||
|
"PropertyDamper",
|
||||||
|
"PropertyMass",
|
||||||
|
"PropertySpring",
|
||||||
|
"SectionProfile",
|
||||||
|
]
|
||||||
@@ -0,0 +1,51 @@
|
|||||||
|
from typing import List
|
||||||
|
|
||||||
|
from ..base import Base
|
||||||
|
from ..geometry import *
|
||||||
|
from .properties import *
|
||||||
|
|
||||||
|
STRUCTURAL_ANALYSIS = "Objects.Structural.Analysis."
|
||||||
|
|
||||||
|
|
||||||
|
class ModelUnits(Base, speckle_type=STRUCTURAL_ANALYSIS + "ModelUnits"):
|
||||||
|
length: str = None
|
||||||
|
sections: str = None
|
||||||
|
displacements: str = None
|
||||||
|
stress: str = None
|
||||||
|
force: str = None
|
||||||
|
mass: str = None
|
||||||
|
time: str = None
|
||||||
|
temperature: str = None
|
||||||
|
velocity: str = None
|
||||||
|
acceleration: str = None
|
||||||
|
energy: str = None
|
||||||
|
angle: str = None
|
||||||
|
strain: str = None
|
||||||
|
|
||||||
|
|
||||||
|
class ModelSettings(Base, speckle_type=STRUCTURAL_ANALYSIS + "ModelSettings"):
|
||||||
|
modelUnits: ModelUnits = None
|
||||||
|
steelCode: str = None
|
||||||
|
concreteCode: str = None
|
||||||
|
coincidenceTolerance: float = 0.0
|
||||||
|
|
||||||
|
|
||||||
|
class ModelInfo(Base, speckle_type=STRUCTURAL_ANALYSIS + "ModelInfo"):
|
||||||
|
name: str = None
|
||||||
|
description: str = None
|
||||||
|
projectNumber: str = None
|
||||||
|
projectName: str = None
|
||||||
|
settings: ModelSettings = None
|
||||||
|
initials: str = None
|
||||||
|
application: str = None
|
||||||
|
|
||||||
|
|
||||||
|
class Model(Base, speckle_type=STRUCTURAL_ANALYSIS + "Model"):
|
||||||
|
specs: ModelInfo = None
|
||||||
|
nodes: List = None
|
||||||
|
elements: List = None
|
||||||
|
loads: List = None
|
||||||
|
restraints: List = None
|
||||||
|
properties: List = None
|
||||||
|
materials: List = None
|
||||||
|
layerDescription: str = None
|
||||||
@@ -0,0 +1,8 @@
|
|||||||
|
from ..base import Base
|
||||||
|
from ..geometry import Plane
|
||||||
|
|
||||||
|
|
||||||
|
class Axis(Base, speckle_type="Objects.Structural.Geometry.Axis"):
|
||||||
|
name: str = None
|
||||||
|
axisType: str = None
|
||||||
|
plane: Plane = None
|
||||||
@@ -0,0 +1,108 @@
|
|||||||
|
from enum import Enum
|
||||||
|
from typing import List
|
||||||
|
|
||||||
|
from ..base import Base
|
||||||
|
from ..geometry import *
|
||||||
|
from .properties import *
|
||||||
|
from .axis import Axis
|
||||||
|
|
||||||
|
STRUCTURAL_GEOMETRY = "Objects.Structural.Geometry"
|
||||||
|
|
||||||
|
|
||||||
|
class ElementType1D(int, Enum):
|
||||||
|
Beam = 0
|
||||||
|
Brace = 1
|
||||||
|
Bar = 2
|
||||||
|
Column = 3
|
||||||
|
Rod = 4
|
||||||
|
Spring = 5
|
||||||
|
Tie = 6
|
||||||
|
Strut = 7
|
||||||
|
Link = 8
|
||||||
|
Damper = 9
|
||||||
|
Cable = 10
|
||||||
|
Spacer = 11
|
||||||
|
Other = 12
|
||||||
|
Null = 13
|
||||||
|
|
||||||
|
|
||||||
|
class ElementType2D(int, Enum):
|
||||||
|
Quad4 = 0
|
||||||
|
Quad8 = 1
|
||||||
|
Triangle3 = 2
|
||||||
|
Triangle6 = 3
|
||||||
|
|
||||||
|
|
||||||
|
class ElementType3D(int, Enum):
|
||||||
|
Brick8 = 0
|
||||||
|
Wedge6 = 1
|
||||||
|
Pyramid5 = 2
|
||||||
|
Tetra4 = 3
|
||||||
|
|
||||||
|
|
||||||
|
class Restraint(Base, speckle_type=STRUCTURAL_GEOMETRY + ".Restraint"):
|
||||||
|
code: str = None
|
||||||
|
stiffnessX: float = 0.0
|
||||||
|
stiffnessY: float = 0.0
|
||||||
|
stiffnessZ: float = 0.0
|
||||||
|
stiffnessXX: float = 0.0
|
||||||
|
stiffnessYY: float = 0.0
|
||||||
|
stiffnessZZ: float = 0.0
|
||||||
|
units: str = None
|
||||||
|
|
||||||
|
|
||||||
|
class Node(Base, speckle_type=STRUCTURAL_GEOMETRY + ".Node"):
|
||||||
|
name: str = None
|
||||||
|
basePoint: Point = None
|
||||||
|
constraintAxis: Axis = None
|
||||||
|
restraint: Restraint = None
|
||||||
|
springProperty: PropertySpring = None
|
||||||
|
massProperty: PropertyMass = None
|
||||||
|
damperProperty: PropertyDamper = None
|
||||||
|
units: str = None
|
||||||
|
|
||||||
|
|
||||||
|
class Element1D(Base, speckle_type=STRUCTURAL_GEOMETRY + ".Element1D"):
|
||||||
|
name: str = None
|
||||||
|
baseLine: Line = None
|
||||||
|
property: Property1D = None
|
||||||
|
type: ElementType1D = None
|
||||||
|
end1Releases: Restraint = None
|
||||||
|
end2Releases: Restraint = None
|
||||||
|
end1Offset: Vector = None
|
||||||
|
end2Offset: Vector = None
|
||||||
|
orientationNode: Node = None
|
||||||
|
orinetationAngle: float = 0.0
|
||||||
|
localAxis: Plane = None
|
||||||
|
parent: Base = None
|
||||||
|
end1Node: Node = Node
|
||||||
|
end2Node: Node = Node
|
||||||
|
topology: List = None
|
||||||
|
displayMesh: Mesh = None
|
||||||
|
units: str = None
|
||||||
|
|
||||||
|
|
||||||
|
class Element2D(Base, speckle_type=STRUCTURAL_GEOMETRY + ".Element2D"):
|
||||||
|
name: str = None
|
||||||
|
property: Property2D = None
|
||||||
|
type: ElementType2D = None
|
||||||
|
offset: float = 0.0
|
||||||
|
orientationAngle: float = 0.0
|
||||||
|
parent: Base = None
|
||||||
|
topology: List = None
|
||||||
|
displayMesh: Mesh = None
|
||||||
|
units: str = None
|
||||||
|
|
||||||
|
|
||||||
|
class Element3D(Base, speckle_type=STRUCTURAL_GEOMETRY + ".Element3D"):
|
||||||
|
name: str = None
|
||||||
|
baseMesh: Mesh = None
|
||||||
|
property: Property3D = None
|
||||||
|
type: ElementType3D = None
|
||||||
|
orientationAngle: float = 0.0
|
||||||
|
parent: Base = None
|
||||||
|
topology: List
|
||||||
|
units: str = None
|
||||||
|
|
||||||
|
|
||||||
|
# class Storey needs ependency on built elements first
|
||||||
@@ -0,0 +1,144 @@
|
|||||||
|
from enum import Enum
|
||||||
|
from typing import List
|
||||||
|
|
||||||
|
from ..base import Base
|
||||||
|
from .geometry import *
|
||||||
|
|
||||||
|
STRUCTURAL_LOADING = "Objects.Structural.Loading."
|
||||||
|
|
||||||
|
|
||||||
|
class LoadType(int, Enum):
|
||||||
|
|
||||||
|
none = 0
|
||||||
|
Dead = 1
|
||||||
|
SuperDead = 2
|
||||||
|
Soil = 3
|
||||||
|
Live = 4
|
||||||
|
LiveRoof = 5
|
||||||
|
ReducibleLive = 6
|
||||||
|
Wind = 7
|
||||||
|
Snow = 8
|
||||||
|
Rain = 9
|
||||||
|
Thermal = 10
|
||||||
|
Notional = 11
|
||||||
|
Prestress = 12
|
||||||
|
Equivalent = 13
|
||||||
|
Accidental = 14
|
||||||
|
SeismicRSA = 15
|
||||||
|
SeismicAccTorsion = 16
|
||||||
|
SeismicStatic = 17
|
||||||
|
Other = 18
|
||||||
|
|
||||||
|
|
||||||
|
class ActionType(int, Enum):
|
||||||
|
|
||||||
|
none = 0
|
||||||
|
Permanent = 1
|
||||||
|
Variable = 2
|
||||||
|
Accidental = 3
|
||||||
|
|
||||||
|
|
||||||
|
class BeamLoadType(int, Enum):
|
||||||
|
|
||||||
|
Point = 0
|
||||||
|
Uniform = 1
|
||||||
|
Linear = 2
|
||||||
|
Patch = 3
|
||||||
|
TriLinear = 4
|
||||||
|
|
||||||
|
|
||||||
|
class FaceLoadType(int, Enum):
|
||||||
|
|
||||||
|
Constant = 0
|
||||||
|
Variable = 1
|
||||||
|
Point = 2
|
||||||
|
|
||||||
|
|
||||||
|
class LoadDirection2D(int, Enum):
|
||||||
|
|
||||||
|
X = 0
|
||||||
|
Y = 1
|
||||||
|
Z = 2
|
||||||
|
|
||||||
|
|
||||||
|
class LoadDirection(int, Enum):
|
||||||
|
|
||||||
|
X = 0
|
||||||
|
Y = 1
|
||||||
|
Z = 2
|
||||||
|
XX = 3
|
||||||
|
YY = 4
|
||||||
|
ZZ = 5
|
||||||
|
|
||||||
|
|
||||||
|
class LoadAxisType(int, Enum):
|
||||||
|
Global = 0
|
||||||
|
Local = 1 # local element axes
|
||||||
|
DeformedLocal = (
|
||||||
|
2 # element local axis that is embedded in the element as it deforms
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class CombinationType(int, Enum):
|
||||||
|
|
||||||
|
LinearAdd = 0
|
||||||
|
Envelope = 1
|
||||||
|
AbsoluteAdd = 2
|
||||||
|
SRSS = 3
|
||||||
|
RangeAdd = 4
|
||||||
|
|
||||||
|
|
||||||
|
class LoadCase(Base, speckle_type=STRUCTURAL_LOADING + "LoadCase"):
|
||||||
|
name: str = None
|
||||||
|
loadType: LoadType = None
|
||||||
|
group: str = None
|
||||||
|
actionType: ActionType = None
|
||||||
|
description: str = None
|
||||||
|
|
||||||
|
|
||||||
|
class Load(Base, speckle_type=STRUCTURAL_LOADING + "Load"):
|
||||||
|
name: str = None
|
||||||
|
units: str = None
|
||||||
|
loadCase: LoadCase = None
|
||||||
|
|
||||||
|
|
||||||
|
class LoadBeam(Load, speckle_type=STRUCTURAL_LOADING + "LoadBeam"):
|
||||||
|
elements: List = None
|
||||||
|
loadType: BeamLoadType = None
|
||||||
|
direction: LoadDirection = None
|
||||||
|
loadAxis: Axis = None
|
||||||
|
loadAxisType: LoadAxisType = None
|
||||||
|
isProjected: bool = None
|
||||||
|
values: List = None
|
||||||
|
positions: List = None
|
||||||
|
|
||||||
|
|
||||||
|
class LoadCombinations(Base, speckle_type=STRUCTURAL_LOADING + "LoadCombination"):
|
||||||
|
name: str = None
|
||||||
|
loadCases: List
|
||||||
|
loadFactors: List
|
||||||
|
combinationType: CombinationType
|
||||||
|
|
||||||
|
|
||||||
|
class LoadFace(Load, speckle_type=STRUCTURAL_LOADING + "LoadFace"):
|
||||||
|
elements: List = None
|
||||||
|
loadType: FaceLoadType = None
|
||||||
|
direction: LoadDirection2D = None
|
||||||
|
loadAxis: Axis = None
|
||||||
|
loadAxisType: LoadAxisType = None
|
||||||
|
isProjected: bool = None
|
||||||
|
values: List = None
|
||||||
|
positions: List = None
|
||||||
|
|
||||||
|
|
||||||
|
class LoadGravity(Load, speckle_type=STRUCTURAL_LOADING + "LoadGravity"):
|
||||||
|
elements: List = None
|
||||||
|
nodes: List = None
|
||||||
|
gravityFactors: Vector = None
|
||||||
|
|
||||||
|
|
||||||
|
class LoadNode(Load, speckle_type=STRUCTURAL_LOADING + "LoadNode"):
|
||||||
|
nodes: List = None
|
||||||
|
loadAxis: Axis = None
|
||||||
|
direction: LoadDirection = None
|
||||||
|
value: float = 0.0
|
||||||
@@ -0,0 +1,59 @@
|
|||||||
|
from enum import Enum
|
||||||
|
|
||||||
|
from ..base import Base
|
||||||
|
|
||||||
|
|
||||||
|
STRUCTURAL_MATERIALS = "Objects.Structural.Materials"
|
||||||
|
|
||||||
|
|
||||||
|
class MaterialType(int, Enum):
|
||||||
|
Concrete = 0
|
||||||
|
Steel = 1
|
||||||
|
Timber = 2
|
||||||
|
Aluminium = 3
|
||||||
|
Masonry = 4
|
||||||
|
FRP = 5
|
||||||
|
Glass = 6
|
||||||
|
Fabric = 7
|
||||||
|
Rebar = 8
|
||||||
|
Tendon = 9
|
||||||
|
ColdFormed = 10
|
||||||
|
Other = 11
|
||||||
|
|
||||||
|
|
||||||
|
class Material(Base, speckle_type=STRUCTURAL_MATERIALS):
|
||||||
|
name: str = None
|
||||||
|
grade: str = None
|
||||||
|
materialType: MaterialType = None
|
||||||
|
designCode: str = None
|
||||||
|
codeYear: str = None
|
||||||
|
strength: float = 0.0
|
||||||
|
elasticModulus: float = 0.0
|
||||||
|
poissonsRatio: float = 0.0
|
||||||
|
shearModulus: float = 0.0
|
||||||
|
density: float = 0.0
|
||||||
|
thermalExpansivity: float = 0.0
|
||||||
|
dampingRatio: float = 0.0
|
||||||
|
cost: float = 0.0
|
||||||
|
materialSafetyFactor: float = 0.0
|
||||||
|
|
||||||
|
|
||||||
|
class Concrete(Material, speckle_type=STRUCTURAL_MATERIALS + ".Concrete"):
|
||||||
|
compressiveStrength: float = 0.0
|
||||||
|
tensileStrength: float = 0.0
|
||||||
|
flexuralStrength: float = 0.0
|
||||||
|
maxCompressiveStrength: float = 0.0
|
||||||
|
maxTensileStrength: float = 0.0
|
||||||
|
maxAggregateSize: float = 0.0
|
||||||
|
lightweight: bool = None
|
||||||
|
|
||||||
|
|
||||||
|
class Steel(Material, speckle_type=STRUCTURAL_MATERIALS + ".Steel"):
|
||||||
|
yieldStrength: float = 0.0
|
||||||
|
ultimateStrength: float = 0.0
|
||||||
|
maxStrain: float = 0.0
|
||||||
|
strainHardeningModulus: float = 0.0
|
||||||
|
|
||||||
|
|
||||||
|
class Timber(Material, speckle_type=STRUCTURAL_MATERIALS + ".Timber"):
|
||||||
|
species: str = None
|
||||||
@@ -0,0 +1,212 @@
|
|||||||
|
from enum import Enum
|
||||||
|
|
||||||
|
from ..base import Base
|
||||||
|
|
||||||
|
from .material import *
|
||||||
|
from .axis import Axis
|
||||||
|
|
||||||
|
|
||||||
|
STRUCTURAL_PROPERTY = "Objectives.Structural.Properties"
|
||||||
|
|
||||||
|
|
||||||
|
class MemberType(int, Enum):
|
||||||
|
Beam = 0
|
||||||
|
Column = 1
|
||||||
|
Generic1D = 2
|
||||||
|
Slab = 3
|
||||||
|
Wall = 4
|
||||||
|
Generic2D = 5
|
||||||
|
VoidCutter1D = 6
|
||||||
|
VoidCutter2D = 7
|
||||||
|
|
||||||
|
|
||||||
|
class BaseReferencePoint(int, Enum):
|
||||||
|
Centroid = 0
|
||||||
|
TopLeft = 1
|
||||||
|
TopCentre = 2
|
||||||
|
TopRight = 3
|
||||||
|
MidLeft = 4
|
||||||
|
MidRight = 5
|
||||||
|
BotLeft = 6
|
||||||
|
BotCentre = 7
|
||||||
|
BotRight = 8
|
||||||
|
|
||||||
|
|
||||||
|
class ReferenceSurface(int, Enum):
|
||||||
|
Top = 0
|
||||||
|
Middle = 1
|
||||||
|
Bottom = 2
|
||||||
|
|
||||||
|
|
||||||
|
class PropertyType2D(int, Enum):
|
||||||
|
|
||||||
|
Stress = 0
|
||||||
|
Fabric = 1
|
||||||
|
Plate = 2
|
||||||
|
Shell = 3
|
||||||
|
Curved = 4
|
||||||
|
Wall = 5
|
||||||
|
Strain = 6
|
||||||
|
Axi = 7
|
||||||
|
Load = 8
|
||||||
|
|
||||||
|
|
||||||
|
class PropertyType3D(int, Enum):
|
||||||
|
Solid = 0
|
||||||
|
Infinite = 1
|
||||||
|
|
||||||
|
|
||||||
|
class ShapeType(int, Enum):
|
||||||
|
Rectangular = 0
|
||||||
|
Circular = 1
|
||||||
|
I = 2
|
||||||
|
Tee = 3
|
||||||
|
Angle = 4
|
||||||
|
Channel = 5
|
||||||
|
Perimeter = 6
|
||||||
|
Box = 7
|
||||||
|
Catalogue = 8
|
||||||
|
Explicit = 9
|
||||||
|
|
||||||
|
|
||||||
|
class PropertyTypeSpring(int, Enum):
|
||||||
|
Axial = 0
|
||||||
|
Torsional = 1
|
||||||
|
General = 2
|
||||||
|
Matrix = 3
|
||||||
|
TensionOnly = 4
|
||||||
|
CompressionOnly = 5
|
||||||
|
Connector = 6
|
||||||
|
LockUp = 7
|
||||||
|
Gap = 8
|
||||||
|
Friction = 9
|
||||||
|
|
||||||
|
|
||||||
|
class PropertyTypeDamper(int, Enum):
|
||||||
|
Axial = 0
|
||||||
|
Torsional = 1
|
||||||
|
General = 2
|
||||||
|
|
||||||
|
|
||||||
|
class Property(Base, speckle_type=STRUCTURAL_PROPERTY):
|
||||||
|
name: str = None
|
||||||
|
|
||||||
|
|
||||||
|
class SectionProfile(Base, speckle_type=STRUCTURAL_PROPERTY + ".SectionProfile"):
|
||||||
|
name: str = None
|
||||||
|
shapeType: ShapeType = None
|
||||||
|
area: float = 0.0
|
||||||
|
Iyy: float = 0.0
|
||||||
|
Izz: float = 0.0
|
||||||
|
J: float = 0.0
|
||||||
|
Ky: float = 0.0
|
||||||
|
weight: float = 0.0
|
||||||
|
units: str = None
|
||||||
|
|
||||||
|
|
||||||
|
class Property1D(Property, speckle_type=STRUCTURAL_PROPERTY + ".Property1D"):
|
||||||
|
memberType: MemberType = None
|
||||||
|
Material: Material = None
|
||||||
|
SectionProfile: SectionProfile = None
|
||||||
|
BaseReferencePoint: BaseReferencePoint = None
|
||||||
|
offsetY: float = 0.0
|
||||||
|
offsetZ: float = 0.0
|
||||||
|
|
||||||
|
|
||||||
|
class Property2D(Property, speckle_type=STRUCTURAL_PROPERTY + ".Property2D"):
|
||||||
|
PropertyType2D: PropertyType2D = None
|
||||||
|
thickness: float = 0.0
|
||||||
|
Material: Material = None
|
||||||
|
axis: Axis = None
|
||||||
|
referenceSurface: ReferenceSurface = None
|
||||||
|
zOffset: float = 0.0
|
||||||
|
modifierInPlane: float = 0.0
|
||||||
|
modifierBending: float = 0.0
|
||||||
|
modifierShear: float = 0.0
|
||||||
|
modifierVolume: float = 0.0
|
||||||
|
|
||||||
|
|
||||||
|
class Property3D(Property, speckle_type=STRUCTURAL_PROPERTY + ".Property3D"):
|
||||||
|
PropertyType3D: PropertyType3D = None
|
||||||
|
Material: Material = None
|
||||||
|
axis: Axis = None
|
||||||
|
|
||||||
|
|
||||||
|
class PropertyDamper(Property, speckle_type=STRUCTURAL_PROPERTY + ".PropertyDamper"):
|
||||||
|
damperType: PropertyTypeDamper = None
|
||||||
|
dampingX: float = 0.0
|
||||||
|
dampingY: float = 0.0
|
||||||
|
dampingZ: float = 0.0
|
||||||
|
dampingXX: float = 0.0
|
||||||
|
dampingYY: float = 0.0
|
||||||
|
dampingZZ: float = 0.0
|
||||||
|
|
||||||
|
|
||||||
|
class PropertyMass(Property, speckle_type=STRUCTURAL_PROPERTY + ".PropertyMass"):
|
||||||
|
mass: float = 0.0
|
||||||
|
inertiaXX: float = 0.0
|
||||||
|
inertiaYY: float = 0.0
|
||||||
|
inertiaZZ: float = 0.0
|
||||||
|
inertiaXY: float = 0.0
|
||||||
|
inertiaYZ: float = 0.0
|
||||||
|
inertiaZX: float = 0.0
|
||||||
|
massModified: bool = None
|
||||||
|
massModifierX: float = 0.0
|
||||||
|
massModifierY: float = 0.0
|
||||||
|
massModifierZ: float = 0.0
|
||||||
|
|
||||||
|
|
||||||
|
class PropertySpring(Property, speckle_type=STRUCTURAL_PROPERTY + ".PropertySpring"):
|
||||||
|
springType: PropertyTypeSpring = None
|
||||||
|
springCurveX: float = 0.0
|
||||||
|
stiffnessX: float = 0.0
|
||||||
|
springCurveY: float = 0.0
|
||||||
|
stiffnessY: float = 0.0
|
||||||
|
springCurveZ: float = 0.0
|
||||||
|
stiffnessZ: float = 0.0
|
||||||
|
springCurveXX: float = 0.0
|
||||||
|
stiffnessXX: float = 0.0
|
||||||
|
springCurveYY: float = 0.0
|
||||||
|
stiffnessYY: float = 0.0
|
||||||
|
springCurveZZ: float = 0.0
|
||||||
|
stiffnessZZ: float = 0.0
|
||||||
|
dampingRatio: float = 0.0
|
||||||
|
dampingX: float = 0.0
|
||||||
|
dampingY: float = 0.0
|
||||||
|
dampingZ: float = 0.0
|
||||||
|
dampingXX: float = 0.0
|
||||||
|
dampingYY: float = 0.0
|
||||||
|
dampingZZ: float = 0.0
|
||||||
|
matrix: float = 0.0
|
||||||
|
postiveLockup: float = 0.0
|
||||||
|
frictionCoefficient: float = 0.0
|
||||||
|
|
||||||
|
|
||||||
|
class ReferenceSurfaceEnum(int, Enum):
|
||||||
|
Concrete = 0
|
||||||
|
Steel = 1
|
||||||
|
Timber = 2
|
||||||
|
Aluminium = 3
|
||||||
|
Masonry = 4
|
||||||
|
FRP = 5
|
||||||
|
Glass = 6
|
||||||
|
Fabric = 7
|
||||||
|
Rebar = 8
|
||||||
|
Tendon = 9
|
||||||
|
ColdFormed = 10
|
||||||
|
Other = 11
|
||||||
|
|
||||||
|
|
||||||
|
class shapeType(int, Enum):
|
||||||
|
Concrete = 0
|
||||||
|
Steel = 1
|
||||||
|
Timber = 2
|
||||||
|
Aluminium = 3
|
||||||
|
Masonry = 4
|
||||||
|
FRP = 5
|
||||||
|
Glass = 6
|
||||||
|
Fabric = 7
|
||||||
|
Rebar = 8
|
||||||
|
Tendon = 9
|
||||||
|
ColdFormed = 10
|
||||||
|
Other = 11
|
||||||
@@ -0,0 +1,174 @@
|
|||||||
|
from typing import List
|
||||||
|
|
||||||
|
from ..base import Base
|
||||||
|
from ..geometry import *
|
||||||
|
from .loading import *
|
||||||
|
from .geometry import *
|
||||||
|
from .analysis import Model
|
||||||
|
|
||||||
|
STRUCTURAL_RESULTS = "Objects.Structural.Results."
|
||||||
|
|
||||||
|
|
||||||
|
class Result(Base, speckle_type=STRUCTURAL_RESULTS + "Result"):
|
||||||
|
resultCase: Base = None
|
||||||
|
permutation: str = None
|
||||||
|
description: str = None
|
||||||
|
|
||||||
|
|
||||||
|
class ResultSet1D(Result, speckle_type=STRUCTURAL_RESULTS + "ResultSet1D"):
|
||||||
|
results1D: List
|
||||||
|
|
||||||
|
|
||||||
|
class Result1D(Result, speckle_type=STRUCTURAL_RESULTS + "Result1D"):
|
||||||
|
element: Element1D = None
|
||||||
|
position: float = 0.0
|
||||||
|
dispX: float = 0.0
|
||||||
|
dispY: float = 0.0
|
||||||
|
dispZ: float = 0.0
|
||||||
|
rotXX: float = 0.0
|
||||||
|
rotYY: float = 0.0
|
||||||
|
rotZZ: float = 0.0
|
||||||
|
forceX: float = 0.0
|
||||||
|
forceY: float = 0.0
|
||||||
|
forceZ: float = 0.0
|
||||||
|
momentXX: float = 0.0
|
||||||
|
momentYY: float = 0.0
|
||||||
|
momentZZ: float = 0.0
|
||||||
|
axialStress: float = 0.0
|
||||||
|
shearStressY: float = 0.0
|
||||||
|
shearStressZ: float = 0.0
|
||||||
|
bendingStressYPos: float = 0.0
|
||||||
|
bendingStressYNeg: float = 0.0
|
||||||
|
bendingStressZPos: float = 0.0
|
||||||
|
bendingStressZNeg: float = 0.0
|
||||||
|
combinedStressMax: float = 0.0
|
||||||
|
combinedStressMin: float = 0.0
|
||||||
|
|
||||||
|
|
||||||
|
class ResultSet2D(Result, speckle_type=STRUCTURAL_RESULTS + "ResultSet2D"):
|
||||||
|
results2D: List
|
||||||
|
|
||||||
|
|
||||||
|
class Result2D(Result, speckle_type=STRUCTURAL_RESULTS + "Result2D"):
|
||||||
|
element: Element2D = None
|
||||||
|
position: List
|
||||||
|
dispX: float = 0.0
|
||||||
|
dispY: float = 0.0
|
||||||
|
dispZ: float = 0.0
|
||||||
|
forceXX: float = 0.0
|
||||||
|
forceYY: float = 0.0
|
||||||
|
forceXY: float = 0.0
|
||||||
|
momentXX: float = 0.0
|
||||||
|
momentYY: float = 0.0
|
||||||
|
momentXY: float = 0.0
|
||||||
|
shearX: float = 0.0
|
||||||
|
shearY: float = 0.0
|
||||||
|
stressTopXX: float = 0.0
|
||||||
|
stressTopYY: float = 0.0
|
||||||
|
stressTopZZ: float = 0.0
|
||||||
|
stressTopXY: float = 0.0
|
||||||
|
stressTopYZ: float = 0.0
|
||||||
|
stressTopZX: float = 0.0
|
||||||
|
stressMidXX: float = 0.0
|
||||||
|
stressMidYY: float = 0.0
|
||||||
|
stressMidZZ: float = 0.0
|
||||||
|
stressMidXY: float = 0.0
|
||||||
|
stressMidYZ: float = 0.0
|
||||||
|
stressMidZX: float = 0.0
|
||||||
|
stressBotXX: float = 0.0
|
||||||
|
stressBotYY: float = 0.0
|
||||||
|
stressBotZZ: float = 0.0
|
||||||
|
stressBotXY: float = 0.0
|
||||||
|
stressBotYZ: float = 0.0
|
||||||
|
stressBotZX: float = 0.0
|
||||||
|
|
||||||
|
|
||||||
|
class ResultSet3D(Result, speckle_type=STRUCTURAL_RESULTS + "ResultSet3D"):
|
||||||
|
results3D: List
|
||||||
|
|
||||||
|
|
||||||
|
class Result3D(Result, speckle_type=STRUCTURAL_RESULTS + "Result3D"):
|
||||||
|
element: Element3D = None
|
||||||
|
position: List
|
||||||
|
dispX: float = 0.0
|
||||||
|
dispY: float = 0.0
|
||||||
|
dispZ: float = 0.0
|
||||||
|
stressXX: float = 0.0
|
||||||
|
stressYY: float = 0.0
|
||||||
|
stressZZ: float = 0.0
|
||||||
|
stressXY: float = 0.0
|
||||||
|
stressYZ: float = 0.0
|
||||||
|
stressZX: float = 0.0
|
||||||
|
|
||||||
|
|
||||||
|
class ResultGlobal(Result, speckle_type=STRUCTURAL_RESULTS + "ResultGlobal"):
|
||||||
|
model: Model = None
|
||||||
|
loadX: float = 0.0
|
||||||
|
loadY: float = 0.0
|
||||||
|
loadZ: float = 0.0
|
||||||
|
loadXX: float = 0.0
|
||||||
|
loadYY: float = 0.0
|
||||||
|
loadZZ: float = 0.0
|
||||||
|
reactionX: float = 0.0
|
||||||
|
reactionY: float = 0.0
|
||||||
|
reactionZ: float = 0.0
|
||||||
|
reactionXX: float = 0.0
|
||||||
|
reactionYY: float = 0.0
|
||||||
|
reactionZZ: float = 0.0
|
||||||
|
mode: float = 0.0
|
||||||
|
frequency: float = 0.0
|
||||||
|
loadFactor: float = 0.0
|
||||||
|
modalStiffness: float = 0.0
|
||||||
|
modalGeoStiffness: float = 0.0
|
||||||
|
effMassX: float = 0.0
|
||||||
|
effMassY: float = 0.0
|
||||||
|
effMassZ: float = 0.0
|
||||||
|
effMassXX: float = 0.0
|
||||||
|
effMassYY: float = 0.0
|
||||||
|
effMassZZ: float = 0.0
|
||||||
|
|
||||||
|
|
||||||
|
class ResultSetNode(Result, speckle_type=STRUCTURAL_RESULTS + "ResultSetNode"):
|
||||||
|
resultsNode: List
|
||||||
|
|
||||||
|
|
||||||
|
class ResultNode(Result, speckle_type=STRUCTURAL_RESULTS + " ResultNode"):
|
||||||
|
node: Node = None
|
||||||
|
dispX: float = 0.0
|
||||||
|
dispY: float = 0.0
|
||||||
|
dispZ: float = 0.0
|
||||||
|
rotXX: float = 0.0
|
||||||
|
rotYY: float = 0.0
|
||||||
|
rotZZ: float = 0.0
|
||||||
|
reactionX: float = 0.0
|
||||||
|
reactionY: float = 0.0
|
||||||
|
reactionZ: float = 0.0
|
||||||
|
reactionXX: float = 0.0
|
||||||
|
reactionYY: float = 0.0
|
||||||
|
reactionZZ: float = 0.0
|
||||||
|
constraintX: float = 0.0
|
||||||
|
constraintY: float = 0.0
|
||||||
|
constraintZ: float = 0.0
|
||||||
|
constraintXX: float = 0.0
|
||||||
|
constraintYY: float = 0.0
|
||||||
|
constraintZZ: float = 0.0
|
||||||
|
velX: float = 0.0
|
||||||
|
velY: float = 0.0
|
||||||
|
velZ: float = 0.0
|
||||||
|
velXX: float = 0.0
|
||||||
|
velYY: float = 0.0
|
||||||
|
velZZ: float = 0.0
|
||||||
|
accX: float = 0.0
|
||||||
|
accY: float = 0.0
|
||||||
|
accZ: float = 0.0
|
||||||
|
accXX: float = 0.0
|
||||||
|
accYY: float = 0.0
|
||||||
|
accZZ: float = 0.0
|
||||||
|
|
||||||
|
|
||||||
|
class ResultSetAll(Base, speckle_type=None):
|
||||||
|
resultSet1D: ResultSet1D = None
|
||||||
|
resultSet2D: ResultSet2D = None
|
||||||
|
resultSet3D: ResultSet3D = None
|
||||||
|
resultsGlobal: ResultGlobal = None
|
||||||
|
resultsNode: ResultSetNode = None
|
||||||
@@ -1,4 +1,5 @@
|
|||||||
from specklepy.logging.exceptions import SpeckleException
|
from warnings import warn
|
||||||
|
from specklepy.logging.exceptions import SpeckleException, SpeckleWarning
|
||||||
|
|
||||||
UNITS = ["mm", "cm", "m", "in", "ft", "yd", "mi"]
|
UNITS = ["mm", "cm", "m", "in", "ft", "yd", "mi"]
|
||||||
|
|
||||||
@@ -15,6 +16,8 @@ UNITS_STRINGS = {
|
|||||||
}
|
}
|
||||||
|
|
||||||
UNITS_ENCODINGS = {
|
UNITS_ENCODINGS = {
|
||||||
|
"none": 0,
|
||||||
|
None: 0,
|
||||||
"mm": 1,
|
"mm": 1,
|
||||||
"cm": 2,
|
"cm": 2,
|
||||||
"m": 3,
|
"m": 3,
|
||||||
@@ -27,6 +30,12 @@ UNITS_ENCODINGS = {
|
|||||||
|
|
||||||
|
|
||||||
def get_units_from_string(unit: str):
|
def get_units_from_string(unit: str):
|
||||||
|
if not isinstance(unit, str):
|
||||||
|
warn(
|
||||||
|
f"Invalid units: expected type str but received {type(unit)} ({unit}). Skipping - no units will be set.",
|
||||||
|
SpeckleWarning,
|
||||||
|
)
|
||||||
|
return
|
||||||
unit = str.lower(unit)
|
unit = str.lower(unit)
|
||||||
for name, alternates in UNITS_STRINGS.items():
|
for name, alternates in UNITS_STRINGS.items():
|
||||||
if unit in alternates:
|
if unit in alternates:
|
||||||
@@ -50,7 +59,5 @@ def get_units_from_encoding(unit: int):
|
|||||||
def get_encoding_from_units(unit: str):
|
def get_encoding_from_units(unit: str):
|
||||||
try:
|
try:
|
||||||
return UNITS_ENCODINGS[unit]
|
return UNITS_ENCODINGS[unit]
|
||||||
except KeyError:
|
except KeyError as e:
|
||||||
raise SpeckleException(
|
raise SpeckleException(message=f"No encoding exists for unit {unit}. Please enter a valid unit to encode (eg {UNITS_ENCODINGS}).") from e
|
||||||
message=f"No encoding exists for unit {unit}. Please enter a valid unit to encode (eg {UNITS_ENCODINGS})."
|
|
||||||
)
|
|
||||||
|
|||||||
@@ -1,12 +1,19 @@
|
|||||||
|
import re
|
||||||
import ujson
|
import ujson
|
||||||
import hashlib
|
import hashlib
|
||||||
import re
|
import warnings
|
||||||
|
|
||||||
from uuid import uuid4
|
from uuid import uuid4
|
||||||
|
from enum import Enum
|
||||||
|
from warnings import warn
|
||||||
from typing import Any, Dict, List, Tuple
|
from typing import Any, Dict, List, Tuple
|
||||||
from specklepy.objects.base import Base, DataChunk
|
from specklepy.objects.base import Base, DataChunk
|
||||||
from specklepy.logging.exceptions import SerializationException, SpeckleException
|
from specklepy.logging.exceptions import (
|
||||||
|
SpeckleException,
|
||||||
|
SpeckleWarning,
|
||||||
|
)
|
||||||
from specklepy.transports.abstract_transport import AbstractTransport
|
from specklepy.transports.abstract_transport import AbstractTransport
|
||||||
|
|
||||||
|
# import for serialization
|
||||||
import specklepy.objects.geometry
|
import specklepy.objects.geometry
|
||||||
import specklepy.objects.other
|
import specklepy.objects.other
|
||||||
|
|
||||||
@@ -17,25 +24,49 @@ def hash_obj(obj: Any) -> str:
|
|||||||
return hashlib.sha256(ujson.dumps(obj).encode()).hexdigest()[:32]
|
return hashlib.sha256(ujson.dumps(obj).encode()).hexdigest()[:32]
|
||||||
|
|
||||||
|
|
||||||
|
def safe_json_loads(obj: str, obj_id=None) -> Any:
|
||||||
|
try:
|
||||||
|
return ujson.loads(obj)
|
||||||
|
except ValueError as err:
|
||||||
|
import json
|
||||||
|
|
||||||
|
warn(
|
||||||
|
f"Failed to deserialise object (id: {obj_id}). This is likely a ujson big int error - falling back to json. \nError: {err}",
|
||||||
|
SpeckleWarning,
|
||||||
|
)
|
||||||
|
return json.loads(obj)
|
||||||
|
|
||||||
|
|
||||||
class BaseObjectSerializer:
|
class BaseObjectSerializer:
|
||||||
read_transport: AbstractTransport
|
read_transport: AbstractTransport
|
||||||
write_transports: List[AbstractTransport]
|
write_transports: List[AbstractTransport]
|
||||||
detach_lineage: List[bool] = [] # tracks depth and whether or not to detach
|
detach_lineage: List[bool] # tracks depth and whether or not to detach
|
||||||
lineage: List[str] = [] # keeps track of hash chain through the object tree
|
lineage: List[str] # keeps track of hash chain through the object tree
|
||||||
family_tree: Dict[str, Dict[str, int]] = {}
|
family_tree: Dict[str, Dict[str, int]]
|
||||||
closure_table: Dict[str, Dict[str, int]] = {}
|
closure_table: Dict[str, Dict[str, int]]
|
||||||
|
deserialized: Dict[str, Base] # holds deserialized objects so objects with same id return the same instance
|
||||||
|
|
||||||
def __init__(
|
def __init__(self, write_transports: List[AbstractTransport] = None, read_transport=None) -> None:
|
||||||
self, write_transports: List[AbstractTransport] = [], read_transport=None
|
self.write_transports = write_transports or []
|
||||||
) -> None:
|
|
||||||
self.write_transports = write_transports
|
|
||||||
self.read_transport = read_transport
|
self.read_transport = read_transport
|
||||||
|
self.detach_lineage = []
|
||||||
|
self.lineage = []
|
||||||
|
self.family_tree = {}
|
||||||
|
self.closure_table = {}
|
||||||
|
self.deserialized = {}
|
||||||
|
|
||||||
def write_json(self, base: Base):
|
def write_json(self, base: Base):
|
||||||
self.__reset_writer()
|
"""Serializes a given base object into a json string
|
||||||
self.detach_lineage = [True]
|
Arguments:
|
||||||
hash, obj = self.traverse_base(base)
|
base {Base} -- the base object to be decomposed and serialized
|
||||||
return hash, ujson.dumps(obj)
|
|
||||||
|
Returns:
|
||||||
|
(str, str) -- a tuple containing the object id of the base object and the serialized object string
|
||||||
|
"""
|
||||||
|
|
||||||
|
obj_id, obj = self.traverse_base(base)
|
||||||
|
|
||||||
|
return obj_id, ujson.dumps(obj)
|
||||||
|
|
||||||
def traverse_base(self, base: Base) -> Tuple[str, Dict]:
|
def traverse_base(self, base: Base) -> Tuple[str, Dict]:
|
||||||
"""Decomposes the given base object and builds a serializable dictionary
|
"""Decomposes the given base object and builds a serializable dictionary
|
||||||
@@ -44,8 +75,23 @@ class BaseObjectSerializer:
|
|||||||
base {Base} -- the base object to be decomposed and serialized
|
base {Base} -- the base object to be decomposed and serialized
|
||||||
|
|
||||||
Returns:
|
Returns:
|
||||||
(str, dict) -- a tuple containing the hash (id) of the base object and the constructed serializable dictionary
|
(str, dict) -- a tuple containing the object id of the base object and the constructed serializable dictionary
|
||||||
"""
|
"""
|
||||||
|
self.__reset_writer()
|
||||||
|
|
||||||
|
if self.write_transports:
|
||||||
|
for wt in self.write_transports:
|
||||||
|
wt.begin_write()
|
||||||
|
|
||||||
|
obj_id, obj = self._traverse_base(base)
|
||||||
|
|
||||||
|
if self.write_transports:
|
||||||
|
for wt in self.write_transports:
|
||||||
|
wt.end_write()
|
||||||
|
|
||||||
|
return obj_id, obj
|
||||||
|
|
||||||
|
def _traverse_base(self, base: Base) -> Tuple[str, Dict]:
|
||||||
if not self.detach_lineage:
|
if not self.detach_lineage:
|
||||||
self.detach_lineage = [True]
|
self.detach_lineage = [True]
|
||||||
|
|
||||||
@@ -60,8 +106,8 @@ class BaseObjectSerializer:
|
|||||||
chunkable = False
|
chunkable = False
|
||||||
detach = False
|
detach = False
|
||||||
|
|
||||||
# skip nulls or props marked to be ignored with "__" or "_"
|
# skip props marked to be ignored with "__" or "_"
|
||||||
if value is None or prop.startswith(("__", "_")):
|
if prop.startswith(("__", "_")):
|
||||||
continue
|
continue
|
||||||
|
|
||||||
# don't prepopulate id as this will mess up hashing
|
# don't prepopulate id as this will mess up hashing
|
||||||
@@ -84,17 +130,22 @@ class BaseObjectSerializer:
|
|||||||
prop.startswith("@") or prop in base._detachable or chunkable
|
prop.startswith("@") or prop in base._detachable or chunkable
|
||||||
)
|
)
|
||||||
|
|
||||||
# 1. handle primitives (ints, floats, strings, and bools)
|
# 1. handle None and primitives (ints, floats, strings, and bools)
|
||||||
if isinstance(value, PRIMITIVES):
|
if value is None or isinstance(value, PRIMITIVES):
|
||||||
object_builder[prop] = value
|
object_builder[prop] = value
|
||||||
continue
|
continue
|
||||||
|
|
||||||
|
# NOTE: for dynamic props, this won't be re-serialised as an enum but as an int
|
||||||
|
if isinstance(value, Enum):
|
||||||
|
object_builder[prop] = value.value
|
||||||
|
continue
|
||||||
|
|
||||||
# 2. handle Base objects
|
# 2. handle Base objects
|
||||||
elif isinstance(value, Base):
|
elif isinstance(value, Base):
|
||||||
child_obj = self.traverse_value(value, detach=detach)
|
child_obj = self.traverse_value(value, detach=detach)
|
||||||
if detach and self.write_transports:
|
if detach and self.write_transports:
|
||||||
ref_hash = child_obj["id"]
|
ref_id = child_obj["id"]
|
||||||
object_builder[prop] = self.detach_helper(ref_hash=ref_hash)
|
object_builder[prop] = self.detach_helper(ref_id=ref_id)
|
||||||
else:
|
else:
|
||||||
object_builder[prop] = child_obj
|
object_builder[prop] = child_obj
|
||||||
|
|
||||||
@@ -113,8 +164,8 @@ class BaseObjectSerializer:
|
|||||||
chunk_refs = []
|
chunk_refs = []
|
||||||
for c in chunks:
|
for c in chunks:
|
||||||
self.detach_lineage.append(detach)
|
self.detach_lineage.append(detach)
|
||||||
ref_hash, _ = self.traverse_base(c)
|
ref_id, _ = self._traverse_base(c)
|
||||||
ref_obj = self.detach_helper(ref_hash=ref_hash)
|
ref_obj = self.detach_helper(ref_id=ref_id)
|
||||||
chunk_refs.append(ref_obj)
|
chunk_refs.append(ref_obj)
|
||||||
object_builder[prop] = chunk_refs
|
object_builder[prop] = chunk_refs
|
||||||
|
|
||||||
@@ -133,20 +184,20 @@ class BaseObjectSerializer:
|
|||||||
}
|
}
|
||||||
object_builder["totalChildrenCount"] = len(closure)
|
object_builder["totalChildrenCount"] = len(closure)
|
||||||
|
|
||||||
hash = hash_obj(object_builder)
|
obj_id = hash_obj(object_builder)
|
||||||
|
|
||||||
object_builder["id"] = hash
|
object_builder["id"] = obj_id
|
||||||
if closure:
|
if closure:
|
||||||
object_builder["__closure"] = self.closure_table[hash] = closure
|
object_builder["__closure"] = self.closure_table[obj_id] = closure
|
||||||
|
|
||||||
# write detached or root objects to transports
|
# write detached or root objects to transports
|
||||||
if detached and self.write_transports:
|
if detached and self.write_transports:
|
||||||
for t in self.write_transports:
|
for t in self.write_transports:
|
||||||
t.save_object(id=hash, serialized_object=ujson.dumps(object_builder))
|
t.save_object(id=obj_id, serialized_object=ujson.dumps(object_builder))
|
||||||
|
|
||||||
del self.lineage[-1]
|
del self.lineage[-1]
|
||||||
|
|
||||||
return hash, object_builder
|
return obj_id, object_builder
|
||||||
|
|
||||||
def traverse_value(self, obj: Any, detach: bool = False) -> Any:
|
def traverse_value(self, obj: Any, detach: bool = False) -> Any:
|
||||||
"""Decomposes a given object and constructs a serializable object or dictionary
|
"""Decomposes a given object and constructs a serializable object or dictionary
|
||||||
@@ -160,6 +211,10 @@ class BaseObjectSerializer:
|
|||||||
if isinstance(obj, PRIMITIVES):
|
if isinstance(obj, PRIMITIVES):
|
||||||
return obj
|
return obj
|
||||||
|
|
||||||
|
# NOTE: for dynamic props, this won't be re-serialised as an enum but as an int
|
||||||
|
if isinstance(obj, Enum):
|
||||||
|
return obj.value
|
||||||
|
|
||||||
elif isinstance(obj, (list, tuple, set)):
|
elif isinstance(obj, (list, tuple, set)):
|
||||||
if not detach:
|
if not detach:
|
||||||
return [self.traverse_value(o) for o in obj]
|
return [self.traverse_value(o) for o in obj]
|
||||||
@@ -168,15 +223,15 @@ class BaseObjectSerializer:
|
|||||||
for o in obj:
|
for o in obj:
|
||||||
if isinstance(o, Base):
|
if isinstance(o, Base):
|
||||||
self.detach_lineage.append(detach)
|
self.detach_lineage.append(detach)
|
||||||
hash, _ = self.traverse_base(o)
|
ref_id, _ = self._traverse_base(o)
|
||||||
detached_list.append(self.detach_helper(ref_hash=hash))
|
detached_list.append(self.detach_helper(ref_id=ref_id))
|
||||||
else:
|
else:
|
||||||
detached_list.append(self.traverse_value(o, detach))
|
detached_list.append(self.traverse_value(o, detach))
|
||||||
return detached_list
|
return detached_list
|
||||||
|
|
||||||
elif isinstance(obj, dict):
|
elif isinstance(obj, dict):
|
||||||
for k, v in obj.items():
|
for k, v in obj.items():
|
||||||
if isinstance(v, PRIMITIVES):
|
if isinstance(v, PRIMITIVES) or v is None:
|
||||||
continue
|
continue
|
||||||
else:
|
else:
|
||||||
obj[k] = self.traverse_value(v)
|
obj[k] = self.traverse_value(v)
|
||||||
@@ -184,24 +239,25 @@ class BaseObjectSerializer:
|
|||||||
|
|
||||||
elif isinstance(obj, Base):
|
elif isinstance(obj, Base):
|
||||||
self.detach_lineage.append(detach)
|
self.detach_lineage.append(detach)
|
||||||
_, base_obj = self.traverse_base(obj)
|
_, base_obj = self._traverse_base(obj)
|
||||||
return base_obj
|
return base_obj
|
||||||
|
|
||||||
else:
|
else:
|
||||||
try:
|
try:
|
||||||
return obj.dict()
|
return obj.dict()
|
||||||
except:
|
except:
|
||||||
SerializationException(
|
warn(
|
||||||
message=f"Failed to handle {type(obj)} in `BaseObjectSerializer.traverse_value`",
|
f"Failed to handle {type(obj)} in `BaseObjectSerializer.traverse_value`",
|
||||||
object=obj,
|
SpeckleWarning,
|
||||||
)
|
)
|
||||||
|
|
||||||
return str(obj)
|
return str(obj)
|
||||||
|
|
||||||
def detach_helper(self, ref_hash: str) -> Dict[str, str]:
|
def detach_helper(self, ref_id: str) -> Dict[str, str]:
|
||||||
"""Helper to keep track of detached objects and their depth in the family tree and create reference objects to place in the parent object
|
"""Helper to keep track of detached objects and their depth in the family tree and create reference objects to place in the parent object
|
||||||
|
|
||||||
Arguments:
|
Arguments:
|
||||||
ref_hash {str} -- the hash of the fully traversed object
|
ref_id {str} -- the id of the fully traversed object
|
||||||
|
|
||||||
Returns:
|
Returns:
|
||||||
dict -- a reference object to be inserted into the given object's parent
|
dict -- a reference object to be inserted into the given object's parent
|
||||||
@@ -210,19 +266,19 @@ class BaseObjectSerializer:
|
|||||||
for parent in self.lineage:
|
for parent in self.lineage:
|
||||||
if parent not in self.family_tree:
|
if parent not in self.family_tree:
|
||||||
self.family_tree[parent] = {}
|
self.family_tree[parent] = {}
|
||||||
if ref_hash not in self.family_tree[parent] or self.family_tree[parent][
|
if ref_id not in self.family_tree[parent] or self.family_tree[parent][
|
||||||
ref_hash
|
ref_id
|
||||||
] > len(self.detach_lineage):
|
] > len(self.detach_lineage):
|
||||||
self.family_tree[parent][ref_hash] = len(self.detach_lineage)
|
self.family_tree[parent][ref_id] = len(self.detach_lineage)
|
||||||
|
|
||||||
return {
|
return {
|
||||||
"referencedId": ref_hash,
|
"referencedId": ref_id,
|
||||||
"speckle_type": "reference",
|
"speckle_type": "reference",
|
||||||
}
|
}
|
||||||
|
|
||||||
def __reset_writer(self) -> None:
|
def __reset_writer(self) -> None:
|
||||||
"""Reinitializes the lineage, and other variables that get used during the json writing process"""
|
"""Reinitializes the lineage, and other variables that get used during the json writing process"""
|
||||||
self.detach_lineage = []
|
self.detach_lineage = [True]
|
||||||
self.lineage = []
|
self.lineage = []
|
||||||
self.family_tree = {}
|
self.family_tree = {}
|
||||||
self.closure_table = {}
|
self.closure_table = {}
|
||||||
@@ -238,7 +294,9 @@ class BaseObjectSerializer:
|
|||||||
"""
|
"""
|
||||||
if not obj_string:
|
if not obj_string:
|
||||||
return None
|
return None
|
||||||
obj = ujson.loads(obj_string)
|
|
||||||
|
self.deserialized = {}
|
||||||
|
obj = safe_json_loads(obj_string)
|
||||||
return self.recompose_base(obj=obj)
|
return self.recompose_base(obj=obj)
|
||||||
|
|
||||||
def recompose_base(self, obj: dict) -> Base:
|
def recompose_base(self, obj: dict) -> Base:
|
||||||
@@ -254,7 +312,10 @@ class BaseObjectSerializer:
|
|||||||
if not obj:
|
if not obj:
|
||||||
return
|
return
|
||||||
if isinstance(obj, str):
|
if isinstance(obj, str):
|
||||||
obj = ujson.loads(obj)
|
obj = safe_json_loads(obj)
|
||||||
|
|
||||||
|
if "id" in obj and obj["id"] in self.deserialized:
|
||||||
|
return self.deserialized[obj["id"]]
|
||||||
|
|
||||||
if "speckle_type" in obj and obj["speckle_type"] == "reference":
|
if "speckle_type" in obj and obj["speckle_type"] == "reference":
|
||||||
obj = self.get_child(obj=obj)
|
obj = self.get_child(obj=obj)
|
||||||
@@ -286,19 +347,25 @@ class BaseObjectSerializer:
|
|||||||
|
|
||||||
# 2. handle referenced child objects
|
# 2. handle referenced child objects
|
||||||
elif "referencedId" in value:
|
elif "referencedId" in value:
|
||||||
ref_hash = value["referencedId"]
|
ref_id = value["referencedId"]
|
||||||
ref_obj_str = self.read_transport.get_object(id=ref_hash)
|
ref_obj_str = self.read_transport.get_object(id=ref_id)
|
||||||
if not ref_obj_str:
|
if ref_obj_str:
|
||||||
raise SpeckleException(
|
ref_obj = safe_json_loads(ref_obj_str, ref_id)
|
||||||
f"Could not find the referenced child object of id `{ref_hash}` in the given read transport: {self.read_transport.name}"
|
base.__setattr__(prop, self.recompose_base(obj=ref_obj))
|
||||||
|
else:
|
||||||
|
warnings.warn(
|
||||||
|
f"Could not find the referenced child object of id `{ref_id}` in the given read transport: {self.read_transport.name}",
|
||||||
|
SpeckleWarning,
|
||||||
)
|
)
|
||||||
ref_obj = ujson.loads(ref_obj_str)
|
base.__setattr__(prop, self.handle_value(value))
|
||||||
base.__setattr__(prop, self.recompose_base(obj=ref_obj))
|
|
||||||
|
|
||||||
# 3. handle all other cases (base objects, lists, and dicts)
|
# 3. handle all other cases (base objects, lists, and dicts)
|
||||||
else:
|
else:
|
||||||
base.__setattr__(prop, self.handle_value(value))
|
base.__setattr__(prop, self.handle_value(value))
|
||||||
|
|
||||||
|
if "id" in obj:
|
||||||
|
self.deserialized[obj["id"]] = base
|
||||||
|
|
||||||
return base
|
return base
|
||||||
|
|
||||||
def handle_value(self, obj: Any):
|
def handle_value(self, obj: Any):
|
||||||
@@ -344,10 +411,13 @@ class BaseObjectSerializer:
|
|||||||
return obj
|
return obj
|
||||||
|
|
||||||
def get_child(self, obj: Dict):
|
def get_child(self, obj: Dict):
|
||||||
ref_hash = obj["referencedId"]
|
ref_id = obj["referencedId"]
|
||||||
ref_obj_str = self.read_transport.get_object(id=ref_hash)
|
ref_obj_str = self.read_transport.get_object(id=ref_id)
|
||||||
if not ref_obj_str:
|
if not ref_obj_str:
|
||||||
raise SpeckleException(
|
warnings.warn(
|
||||||
f"Could not find the referenced child object of id `{ref_hash}` in the given read transport: {self.read_transport.name}"
|
f"Could not find the referenced child object of id `{ref_id}` in the given read transport: {self.read_transport.name}",
|
||||||
|
SpeckleWarning,
|
||||||
)
|
)
|
||||||
return ujson.loads(ref_obj_str)
|
return obj
|
||||||
|
|
||||||
|
return safe_json_loads(ref_obj_str, ref_id)
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
from abc import ABC, abstractmethod
|
from abc import ABC, abstractmethod
|
||||||
from typing import Any, Optional, List, Dict
|
from typing import Optional, List, Dict
|
||||||
from pydantic import BaseModel
|
from pydantic import BaseModel
|
||||||
from pydantic.main import Extra
|
from pydantic.main import Extra
|
||||||
|
|
||||||
|
|||||||
@@ -1,6 +1,4 @@
|
|||||||
import json
|
|
||||||
from typing import Any, List, Dict
|
from typing import Any, List, Dict
|
||||||
from specklepy.logging.exceptions import SpeckleException
|
|
||||||
from specklepy.transports.abstract_transport import AbstractTransport
|
from specklepy.transports.abstract_transport import AbstractTransport
|
||||||
|
|
||||||
|
|
||||||
@@ -28,10 +26,7 @@ class MemoryTransport(AbstractTransport):
|
|||||||
raise NotImplementedError
|
raise NotImplementedError
|
||||||
|
|
||||||
def get_object(self, id: str) -> str or None:
|
def get_object(self, id: str) -> str or None:
|
||||||
if id in self.objects:
|
return self.objects[id] if id in self.objects else None
|
||||||
return self.objects[id]
|
|
||||||
else:
|
|
||||||
return None
|
|
||||||
|
|
||||||
def has_objects(self, id_list: List[str]) -> Dict[str, bool]:
|
def has_objects(self, id_list: List[str]) -> Dict[str, bool]:
|
||||||
return {id: (id in self.objects) for id in id_list}
|
return {id: (id in self.objects) for id in id_list}
|
||||||
|
|||||||
@@ -2,7 +2,6 @@ import json
|
|||||||
import logging
|
import logging
|
||||||
import threading
|
import threading
|
||||||
import queue
|
import queue
|
||||||
import time
|
|
||||||
import gzip
|
import gzip
|
||||||
|
|
||||||
import requests
|
import requests
|
||||||
@@ -92,10 +91,15 @@ class BatchSender(object):
|
|||||||
|
|
||||||
def _bg_send_batch(self, session, batch):
|
def _bg_send_batch(self, session, batch):
|
||||||
object_ids = [obj[0] for obj in batch]
|
object_ids = [obj[0] for obj in batch]
|
||||||
server_has_object = session.post(
|
try:
|
||||||
url=f"{self.server_url}/api/diff/{self.stream_id}",
|
server_has_object = session.post(
|
||||||
data={"objects": json.dumps(object_ids)},
|
url=f"{self.server_url}/api/diff/{self.stream_id}",
|
||||||
).json()
|
data={"objects": json.dumps(object_ids)},
|
||||||
|
).json()
|
||||||
|
except Exception as ex:
|
||||||
|
raise SpeckleException(
|
||||||
|
f"Invalid credentials - cannot send objects to server {self.server_url}"
|
||||||
|
) from ex
|
||||||
|
|
||||||
new_object_ids = [x for x in object_ids if not server_has_object[x]]
|
new_object_ids = [x for x in object_ids if not server_has_object[x]]
|
||||||
new_object_ids = set(new_object_ids)
|
new_object_ids = set(new_object_ids)
|
||||||
|
|||||||
@@ -1,40 +1,97 @@
|
|||||||
import json
|
import json
|
||||||
import time
|
|
||||||
|
|
||||||
import requests
|
import requests
|
||||||
|
from warnings import warn
|
||||||
|
|
||||||
from typing import Any, Dict, List, Type
|
from typing import Any, Dict, List
|
||||||
|
|
||||||
from specklepy.api.client import SpeckleClient
|
from specklepy.api.client import SpeckleClient
|
||||||
from specklepy.logging.exceptions import SpeckleException
|
from specklepy.api.credentials import Account, get_account_from_token
|
||||||
|
from specklepy.logging.exceptions import SpeckleException, SpeckleWarning
|
||||||
from specklepy.transports.abstract_transport import AbstractTransport
|
from specklepy.transports.abstract_transport import AbstractTransport
|
||||||
|
|
||||||
from .batch_sender import BatchSender
|
from .batch_sender import BatchSender
|
||||||
|
|
||||||
|
|
||||||
class ServerTransport(AbstractTransport):
|
class ServerTransport(AbstractTransport):
|
||||||
|
"""
|
||||||
|
The `ServerTransport` is the vehicle through which you transport objects to and from a Speckle Server. Provide it to
|
||||||
|
`operations.send()` or `operations.receive()`.
|
||||||
|
|
||||||
|
The `ServerTransport` can be authenticted two different ways:
|
||||||
|
1. by providing a `SpeckleClient`
|
||||||
|
2. by providing an `Account`
|
||||||
|
3. by providing a `token` and `url`
|
||||||
|
|
||||||
|
```py
|
||||||
|
from specklepy.api import operations
|
||||||
|
from specklepy.transports.server import ServerTransport
|
||||||
|
|
||||||
|
# here's the data you want to send
|
||||||
|
block = Block(length=2, height=4)
|
||||||
|
|
||||||
|
# next create the server transport - this is the vehicle through which you will send and receive
|
||||||
|
transport = ServerTransport(stream_id=new_stream_id, client=client)
|
||||||
|
|
||||||
|
# this serialises the block and sends it to the transport
|
||||||
|
hash = operations.send(base=block, transports=[transport])
|
||||||
|
|
||||||
|
# you can now create a commit on your stream with this object
|
||||||
|
commid_id = client.commit.create(
|
||||||
|
stream_id=new_stream_id,
|
||||||
|
obj_id=hash,
|
||||||
|
message="this is a block I made in speckle-py",
|
||||||
|
)
|
||||||
|
```
|
||||||
|
"""
|
||||||
|
|
||||||
_name = "RemoteTransport"
|
_name = "RemoteTransport"
|
||||||
url: str = None
|
url: str = None
|
||||||
stream_id: str = None
|
stream_id: str = None
|
||||||
|
account: Account = None
|
||||||
saved_obj_count: int = 0
|
saved_obj_count: int = 0
|
||||||
session: requests.Session = None
|
session: requests.Session = None
|
||||||
|
|
||||||
def __init__(self, client: SpeckleClient, stream_id: str, **data: Any) -> None:
|
def __init__(
|
||||||
|
self,
|
||||||
|
stream_id: str,
|
||||||
|
client: SpeckleClient = None,
|
||||||
|
account: Account = None,
|
||||||
|
token: str = None,
|
||||||
|
url: str = None,
|
||||||
|
**data: Any,
|
||||||
|
) -> None:
|
||||||
super().__init__(**data)
|
super().__init__(**data)
|
||||||
# TODO: replace client with account or some other auth avenue
|
if client is None and account is None and token is None and url is None:
|
||||||
if not client.me:
|
raise SpeckleException(
|
||||||
raise SpeckleException("The provided SpeckleClient was not authenticated.")
|
"You must provide either a client or a token and url to construct a ServerTransport."
|
||||||
self.url = client.url
|
)
|
||||||
self.stream_id = stream_id
|
|
||||||
|
if account:
|
||||||
|
self.account = account
|
||||||
|
url = account.serverInfo.url
|
||||||
|
elif client:
|
||||||
|
url = client.url
|
||||||
|
if not client.account.token:
|
||||||
|
warn(
|
||||||
|
SpeckleWarning(
|
||||||
|
f"Unauthenticated Speckle Client provided to Server Transport for {self.url}. Receiving from private streams will fail."
|
||||||
|
)
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
self.account = client.account
|
||||||
|
else:
|
||||||
|
self.account = get_account_from_token(token, url)
|
||||||
|
|
||||||
|
self.stream_id = stream_id
|
||||||
|
self.url = url
|
||||||
|
|
||||||
token = client.me["token"]
|
|
||||||
self._batch_sender = BatchSender(
|
self._batch_sender = BatchSender(
|
||||||
self.url, self.stream_id, token, max_batch_size_mb=1
|
self.url, self.stream_id, self.account.token, max_batch_size_mb=1
|
||||||
)
|
)
|
||||||
|
|
||||||
self.session = requests.Session()
|
self.session = requests.Session()
|
||||||
self.session.headers.update(
|
self.session.headers.update(
|
||||||
{"Authorization": f"Bearer {token}", "Accept": "text/plain"}
|
{"Authorization": f"Bearer {self.account.token}", "Accept": "text/plain"}
|
||||||
)
|
)
|
||||||
|
|
||||||
def begin_write(self) -> None:
|
def begin_write(self) -> None:
|
||||||
@@ -73,8 +130,7 @@ class ServerTransport(AbstractTransport):
|
|||||||
) -> str:
|
) -> str:
|
||||||
endpoint = f"{self.url}/objects/{self.stream_id}/{id}/single"
|
endpoint = f"{self.url}/objects/{self.stream_id}/{id}/single"
|
||||||
r = self.session.get(endpoint)
|
r = self.session.get(endpoint)
|
||||||
if r.encoding is None:
|
r.encoding = "utf-8"
|
||||||
r.encoding = "utf-8"
|
|
||||||
|
|
||||||
if r.status_code != 200:
|
if r.status_code != 200:
|
||||||
raise SpeckleException(
|
raise SpeckleException(
|
||||||
@@ -96,34 +152,17 @@ class ServerTransport(AbstractTransport):
|
|||||||
r = self.session.post(
|
r = self.session.post(
|
||||||
endpoint, data={"objects": json.dumps(new_children_ids)}, stream=True
|
endpoint, data={"objects": json.dumps(new_children_ids)}, stream=True
|
||||||
)
|
)
|
||||||
if r.encoding is None:
|
r.encoding = "utf-8"
|
||||||
r.encoding = "utf-8"
|
|
||||||
lines = r.iter_lines(decode_unicode=True)
|
lines = r.iter_lines(decode_unicode=True)
|
||||||
|
|
||||||
# iter through returned objects saving them as we go
|
# iter through returned objects saving them as we go
|
||||||
|
target_transport.begin_write()
|
||||||
for line in lines:
|
for line in lines:
|
||||||
if line:
|
if line:
|
||||||
hash, obj = line.split("\t")
|
hash, obj = line.split("\t")
|
||||||
target_transport.save_object(hash, obj)
|
target_transport.save_object(hash, obj)
|
||||||
|
|
||||||
target_transport.save_object(id, root_obj_serialized)
|
target_transport.save_object(id, root_obj_serialized)
|
||||||
|
target_transport.end_write()
|
||||||
|
|
||||||
return root_obj_serialized
|
return root_obj_serialized
|
||||||
|
|
||||||
# async def stream_res(self, endpoint: str) -> str:
|
|
||||||
# data = b""
|
|
||||||
# async with aiohttp.ClientSession() as session:
|
|
||||||
# session.headers.update(
|
|
||||||
# {
|
|
||||||
# "Authorization": f"{self.session.headers['Authorization']}",
|
|
||||||
# "Accept": "text/plain",
|
|
||||||
# }
|
|
||||||
# )
|
|
||||||
# async with session.get(endpoint) as res:
|
|
||||||
# while True:
|
|
||||||
# chunk = await res.content.read(self.chunk_size)
|
|
||||||
# if not chunk:
|
|
||||||
# break
|
|
||||||
# data += chunk
|
|
||||||
|
|
||||||
# return data.decode("utf-8")
|
|
||||||
|
|||||||
@@ -3,7 +3,7 @@ import sys
|
|||||||
import time
|
import time
|
||||||
import sched
|
import sched
|
||||||
import sqlite3
|
import sqlite3
|
||||||
from typing import Any, List, Dict
|
from typing import Any, List, Dict, Tuple
|
||||||
from appdirs import user_data_dir
|
from appdirs import user_data_dir
|
||||||
from contextlib import closing
|
from contextlib import closing
|
||||||
from specklepy.transports.abstract_transport import AbstractTransport
|
from specklepy.transports.abstract_transport import AbstractTransport
|
||||||
@@ -14,25 +14,29 @@ class SQLiteTransport(AbstractTransport):
|
|||||||
_name = "SQLite"
|
_name = "SQLite"
|
||||||
_base_path: str = None
|
_base_path: str = None
|
||||||
_root_path: str = None
|
_root_path: str = None
|
||||||
_is_writing: bool = False
|
|
||||||
_scheduler = sched.scheduler(time.time, time.sleep)
|
|
||||||
_polling_interval = 0.5 # seconds
|
|
||||||
__connection: sqlite3.Connection = None
|
__connection: sqlite3.Connection = None
|
||||||
app_name: str = ""
|
app_name: str = ""
|
||||||
scope: str = ""
|
scope: str = ""
|
||||||
saved_obj_count: int = 0
|
saved_obj_count: int = 0
|
||||||
|
max_size: int = None
|
||||||
|
_current_batch: List[Tuple[str, str]] = None
|
||||||
|
_current_batch_size: int = None
|
||||||
|
|
||||||
def __init__(
|
def __init__(
|
||||||
self,
|
self,
|
||||||
base_path: str = None,
|
base_path: str = None,
|
||||||
app_name: str = None,
|
app_name: str = None,
|
||||||
scope: str = None,
|
scope: str = None,
|
||||||
|
max_batch_size_mb: float = 10.0,
|
||||||
**data: Any,
|
**data: Any,
|
||||||
) -> None:
|
) -> None:
|
||||||
super().__init__(**data)
|
super().__init__(**data)
|
||||||
self.app_name = app_name or "Speckle"
|
self.app_name = app_name or "Speckle"
|
||||||
self.scope = scope or "Objects"
|
self.scope = scope or "Objects"
|
||||||
self._base_path = base_path or self.__get_base_path()
|
self._base_path = base_path or self.get_base_path(self.app_name)
|
||||||
|
self.max_size = int(max_batch_size_mb * 1000 * 1000)
|
||||||
|
self._current_batch = []
|
||||||
|
self._current_batch_size = 0
|
||||||
|
|
||||||
try:
|
try:
|
||||||
os.makedirs(self._base_path, exist_ok=True)
|
os.makedirs(self._base_path, exist_ok=True)
|
||||||
@@ -50,13 +54,8 @@ class SQLiteTransport(AbstractTransport):
|
|||||||
def __repr__(self) -> str:
|
def __repr__(self) -> str:
|
||||||
return f"SQLiteTransport(app: '{self.app_name}', scope: '{self.scope}')"
|
return f"SQLiteTransport(app: '{self.app_name}', scope: '{self.scope}')"
|
||||||
|
|
||||||
# def __write_timer_elapsed(self):
|
@staticmethod
|
||||||
# print("WRITE TIMER ELAPSED")
|
def get_base_path(app_name):
|
||||||
# proc = Process(target=_run_queue, args=(self.__queue, self._root_path))
|
|
||||||
# proc.start()
|
|
||||||
# proc.join()
|
|
||||||
|
|
||||||
def __get_base_path(self):
|
|
||||||
# from appdirs https://github.com/ActiveState/appdirs/blob/master/appdirs.py
|
# from appdirs https://github.com/ActiveState/appdirs/blob/master/appdirs.py
|
||||||
# default mac path is not the one we use (we use unix path), so using special case for this
|
# default mac path is not the one we use (we use unix path), so using special case for this
|
||||||
system = sys.platform
|
system = sys.platform
|
||||||
@@ -68,40 +67,10 @@ class SQLiteTransport(AbstractTransport):
|
|||||||
system = "darwin"
|
system = "darwin"
|
||||||
|
|
||||||
if system != "darwin":
|
if system != "darwin":
|
||||||
return user_data_dir(appname=self.app_name, appauthor=False, roaming=True)
|
return user_data_dir(appname=app_name, appauthor=False, roaming=True)
|
||||||
|
|
||||||
path = os.path.expanduser("~/.config/")
|
path = os.path.expanduser("~/.config/")
|
||||||
return os.path.join(path, self.app_name)
|
return os.path.join(path, app_name)
|
||||||
|
|
||||||
# def __consume_queue(self):
|
|
||||||
# if self._is_writing or self.__queue.empty():
|
|
||||||
# return
|
|
||||||
# print("CONSUME QUEUE")
|
|
||||||
# self._is_writing = True
|
|
||||||
# while not self.__queue.empty():
|
|
||||||
# data = self.__queue.get()
|
|
||||||
# self.save_object(data[0], data[1])
|
|
||||||
# self._is_writing = False
|
|
||||||
|
|
||||||
# self._scheduler.enter(
|
|
||||||
# delay=self._polling_interval, priority=1, action=self.__consume_queue
|
|
||||||
# )
|
|
||||||
# self._scheduler.run(blocking=True)
|
|
||||||
|
|
||||||
# def save_object(self, id: str, serialized_object: str) -> None:
|
|
||||||
# """Adds an object to the queue and schedules it to be saved.
|
|
||||||
|
|
||||||
# Arguments:
|
|
||||||
# id {str} -- the object id
|
|
||||||
# serialized_object {str} -- the full string representation of the object
|
|
||||||
# """
|
|
||||||
# print("SAVE OBJECT")
|
|
||||||
# self.__queue.put((id, serialized_object))
|
|
||||||
|
|
||||||
# self._scheduler.enter(
|
|
||||||
# delay=self._polling_interval, priority=1, action=self.__consume_queue
|
|
||||||
# )
|
|
||||||
# self._scheduler.run(blocking=True)
|
|
||||||
|
|
||||||
def save_object_from_transport(
|
def save_object_from_transport(
|
||||||
self, id: str, source_transport: AbstractTransport
|
self, id: str, source_transport: AbstractTransport
|
||||||
@@ -116,23 +85,41 @@ class SQLiteTransport(AbstractTransport):
|
|||||||
self.save_object(id, serialized_object)
|
self.save_object(id, serialized_object)
|
||||||
|
|
||||||
def save_object(self, id: str, serialized_object: str) -> None:
|
def save_object(self, id: str, serialized_object: str) -> None:
|
||||||
"""Directly saves an object into the database.
|
"""
|
||||||
|
Adds an object to the current batch to be written to the db. If the current batch is full,
|
||||||
|
the batch is written to the db and the current batch is reset.
|
||||||
|
|
||||||
Arguments:
|
Arguments:
|
||||||
id {str} -- the object id
|
id {str} -- the object id
|
||||||
serialized_object {str} -- the full string representation of the object
|
serialized_object {str} -- the full string representation of the object
|
||||||
"""
|
"""
|
||||||
|
obj_size = len(serialized_object)
|
||||||
|
if (
|
||||||
|
not self._current_batch
|
||||||
|
or self._current_batch_size + obj_size < self.max_size
|
||||||
|
):
|
||||||
|
self._current_batch.append((id, serialized_object))
|
||||||
|
self._current_batch_size += obj_size
|
||||||
|
return
|
||||||
|
|
||||||
|
self.save_current_batch()
|
||||||
|
self._current_batch = [(id, serialized_object)]
|
||||||
|
self._current_batch_size = obj_size
|
||||||
|
|
||||||
|
def save_current_batch(self) -> None:
|
||||||
|
"""Save the current batch of objects to the local db"""
|
||||||
self.__check_connection()
|
self.__check_connection()
|
||||||
try:
|
try:
|
||||||
with closing(self.__connection.cursor()) as c:
|
with closing(self.__connection.cursor()) as c:
|
||||||
c.execute(
|
c.executemany(
|
||||||
"INSERT OR IGNORE INTO objects(hash, content) VALUES(?,?)",
|
"INSERT OR IGNORE INTO objects(hash, content) VALUES(?,?)",
|
||||||
(id, serialized_object),
|
self._current_batch,
|
||||||
)
|
)
|
||||||
self.__connection.commit()
|
self.__connection.commit()
|
||||||
except Exception as ex:
|
except Exception as ex:
|
||||||
raise SpeckleException(
|
raise SpeckleException(
|
||||||
f"Could not save the object to the local db. Inner exception: {ex}", ex
|
f"Could not save the batch of objects to the local db. Inner exception: {ex}",
|
||||||
|
ex,
|
||||||
)
|
)
|
||||||
|
|
||||||
def get_object(self, id: str) -> str or None:
|
def get_object(self, id: str) -> str or None:
|
||||||
@@ -155,10 +142,14 @@ class SQLiteTransport(AbstractTransport):
|
|||||||
return ret
|
return ret
|
||||||
|
|
||||||
def begin_write(self):
|
def begin_write(self):
|
||||||
|
self._object_cache = []
|
||||||
self.saved_obj_count = 0
|
self.saved_obj_count = 0
|
||||||
|
|
||||||
def end_write(self):
|
def end_write(self):
|
||||||
pass
|
if self._current_batch:
|
||||||
|
self.save_current_batch()
|
||||||
|
self._current_batch = []
|
||||||
|
self._current_batch_size = 0
|
||||||
|
|
||||||
def copy_object_and_children(
|
def copy_object_and_children(
|
||||||
self, id: str, target_transport: AbstractTransport
|
self, id: str, target_transport: AbstractTransport
|
||||||
@@ -197,20 +188,4 @@ class SQLiteTransport(AbstractTransport):
|
|||||||
self.__connection = sqlite3.connect(self._root_path)
|
self.__connection = sqlite3.connect(self._root_path)
|
||||||
|
|
||||||
def __del__(self):
|
def __del__(self):
|
||||||
self.__connection.close()
|
self.close()
|
||||||
|
|
||||||
|
|
||||||
# def _run_queue(queue: Queue, root_path: str):
|
|
||||||
# if queue.empty():
|
|
||||||
# return
|
|
||||||
# print("RUN QUEUE")
|
|
||||||
# conn = sqlite3.connect(root_path)
|
|
||||||
# while not queue.empty():
|
|
||||||
# data = queue.get()
|
|
||||||
# with closing(conn.cursor()) as c:
|
|
||||||
# c.execute(
|
|
||||||
# "INSERT OR IGNORE INTO objects(hash, content) VALUES(?,?)",
|
|
||||||
# (data[0], data[1]),
|
|
||||||
# )
|
|
||||||
# conn.commit()
|
|
||||||
# conn.close()
|
|
||||||
|
|||||||
+18
-3
@@ -6,7 +6,10 @@ from specklepy.api.models import Stream
|
|||||||
from specklepy.api.client import SpeckleClient
|
from specklepy.api.client import SpeckleClient
|
||||||
from specklepy.objects.base import Base
|
from specklepy.objects.base import Base
|
||||||
from specklepy.objects.geometry import Point
|
from specklepy.objects.geometry import Point
|
||||||
from specklepy.objects.fakemesh import FakeMesh
|
from specklepy.objects.fakemesh import FakeDirection, FakeMesh
|
||||||
|
from specklepy.logging import metrics
|
||||||
|
|
||||||
|
metrics.disable()
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture(scope="session")
|
@pytest.fixture(scope="session")
|
||||||
@@ -58,7 +61,14 @@ def second_user_dict(host):
|
|||||||
@pytest.fixture(scope="session")
|
@pytest.fixture(scope="session")
|
||||||
def client(host, user_dict):
|
def client(host, user_dict):
|
||||||
client = SpeckleClient(host=host, use_ssl=False)
|
client = SpeckleClient(host=host, use_ssl=False)
|
||||||
client.authenticate(user_dict["token"])
|
client.authenticate_with_token(user_dict["token"])
|
||||||
|
return client
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture(scope="session")
|
||||||
|
def second_client(host, second_user_dict):
|
||||||
|
client = SpeckleClient(host=host, use_ssl=False)
|
||||||
|
client.authenticate_with_token(second_user_dict["token"])
|
||||||
return client
|
return client
|
||||||
|
|
||||||
|
|
||||||
@@ -78,9 +88,10 @@ def mesh():
|
|||||||
mesh = FakeMesh()
|
mesh = FakeMesh()
|
||||||
mesh.name = "my_mesh"
|
mesh.name = "my_mesh"
|
||||||
mesh.vertices = [random.uniform(0, 10) for _ in range(1, 210)]
|
mesh.vertices = [random.uniform(0, 10) for _ in range(1, 210)]
|
||||||
mesh.faces = [i for i in range(1, 210)]
|
mesh.faces = list(range(1, 210))
|
||||||
mesh["@(100)colours"] = [random.uniform(0, 10) for _ in range(1, 210)]
|
mesh["@(100)colours"] = [random.uniform(0, 10) for _ in range(1, 210)]
|
||||||
mesh["@()default_chunk"] = [random.uniform(0, 10) for _ in range(1, 210)]
|
mesh["@()default_chunk"] = [random.uniform(0, 10) for _ in range(1, 210)]
|
||||||
|
mesh.cardinal_dir = FakeDirection.WEST
|
||||||
mesh.test_bases = [Base(name=f"test {i}") for i in range(1, 22)]
|
mesh.test_bases = [Base(name=f"test {i}") for i in range(1, 22)]
|
||||||
mesh.detach_this = Base(name="predefined detached base")
|
mesh.detach_this = Base(name="predefined detached base")
|
||||||
mesh["@detach"] = Base(name="detached base")
|
mesh["@detach"] = Base(name="detached base")
|
||||||
@@ -99,6 +110,10 @@ def base():
|
|||||||
base = Base()
|
base = Base()
|
||||||
base.name = "my_base"
|
base.name = "my_base"
|
||||||
base.units = "millimetres"
|
base.units = "millimetres"
|
||||||
|
base.null_val = None
|
||||||
|
base.null_dict = {"a null val": None}
|
||||||
|
base.tuple = (1, 2, "3")
|
||||||
|
base.set = {1, 2, "3"}
|
||||||
base.vertices = [random.uniform(0, 10) for _ in range(1, 120)]
|
base.vertices = [random.uniform(0, 10) for _ in range(1, 120)]
|
||||||
base.test_bases = [Base(name=i) for i in range(1, 22)]
|
base.test_bases = [Base(name=i) for i in range(1, 22)]
|
||||||
base["@detach"] = Base(name="detached base")
|
base["@detach"] = Base(name="detached base")
|
||||||
|
|||||||
+45
-3
@@ -1,5 +1,6 @@
|
|||||||
|
from enum import Enum
|
||||||
|
from typing import Dict, List, Optional, Union
|
||||||
from contextlib import ExitStack as does_not_raise
|
from contextlib import ExitStack as does_not_raise
|
||||||
from typing import Dict, List
|
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
from specklepy.api import operations
|
from specklepy.api import operations
|
||||||
@@ -77,20 +78,40 @@ def test_speckle_type_cannot_be_set(base: Base) -> None:
|
|||||||
assert base.speckle_type == "Base"
|
assert base.speckle_type == "Base"
|
||||||
|
|
||||||
|
|
||||||
|
def test_setting_units():
|
||||||
|
b = Base(units="foot")
|
||||||
|
assert b.units == "ft"
|
||||||
|
|
||||||
|
with pytest.raises(SpeckleException):
|
||||||
|
b.units = "big"
|
||||||
|
|
||||||
|
b.units = None # invalid args are skipped
|
||||||
|
b.units = 7
|
||||||
|
assert b.units == "ft"
|
||||||
|
|
||||||
|
|
||||||
def test_base_of_custom_speckle_type() -> None:
|
def test_base_of_custom_speckle_type() -> None:
|
||||||
b1 = Base.of_type("BirdHouse", name="Tweety's Crib")
|
b1 = Base.of_type("BirdHouse", name="Tweety's Crib")
|
||||||
assert b1.speckle_type == "BirdHouse"
|
assert b1.speckle_type == "BirdHouse"
|
||||||
assert b1.name == "Tweety's Crib"
|
assert b1.name == "Tweety's Crib"
|
||||||
|
|
||||||
|
|
||||||
|
class DietaryRestrictions(Enum):
|
||||||
|
VEGAN = 1
|
||||||
|
GLUTEN_FREE = 2
|
||||||
|
NUT_FREE = 3
|
||||||
|
|
||||||
|
|
||||||
class FrozenYoghurt(Base):
|
class FrozenYoghurt(Base):
|
||||||
"""Testing type checking"""
|
"""Testing type checking"""
|
||||||
|
|
||||||
servings: int
|
servings: int
|
||||||
flavours: List[str] = None # list item types won't be checked
|
flavours: List[str] # list item types won't be checked
|
||||||
customer: str
|
customer: str
|
||||||
add_ons: Dict[str, float] # dict item types won't be checked
|
add_ons: Optional[Dict[str, float]] # dict item types won't be checked
|
||||||
price: float = 0.0
|
price: float = 0.0
|
||||||
|
dietary: DietaryRestrictions
|
||||||
|
tag: Union[int, str]
|
||||||
|
|
||||||
|
|
||||||
def test_type_checking() -> None:
|
def test_type_checking() -> None:
|
||||||
@@ -99,6 +120,9 @@ def test_type_checking() -> None:
|
|||||||
order.servings = 2
|
order.servings = 2
|
||||||
order.price = "7" # will get converted
|
order.price = "7" # will get converted
|
||||||
order.customer = "izzy"
|
order.customer = "izzy"
|
||||||
|
order.dietary = DietaryRestrictions.VEGAN
|
||||||
|
order.tag = "preorder"
|
||||||
|
order.tag = 4411
|
||||||
|
|
||||||
with pytest.raises(SpeckleException):
|
with pytest.raises(SpeckleException):
|
||||||
order.flavours = "not a list"
|
order.flavours = "not a list"
|
||||||
@@ -106,6 +130,10 @@ def test_type_checking() -> None:
|
|||||||
order.servings = "five"
|
order.servings = "five"
|
||||||
with pytest.raises(SpeckleException):
|
with pytest.raises(SpeckleException):
|
||||||
order.add_ons = ["sprinkles"]
|
order.add_ons = ["sprinkles"]
|
||||||
|
with pytest.raises(SpeckleException):
|
||||||
|
order.dietary = "no nuts plz"
|
||||||
|
with pytest.raises(SpeckleException):
|
||||||
|
order.tag = ["tag01", "tag02"]
|
||||||
|
|
||||||
order.add_ons = {"sprinkles": 0.2, "chocolate": 1.0}
|
order.add_ons = {"sprinkles": 0.2, "chocolate": 1.0}
|
||||||
order.flavours = ["strawberry", "lychee", "peach", "pineapple"]
|
order.flavours = ["strawberry", "lychee", "peach", "pineapple"]
|
||||||
@@ -113,3 +141,17 @@ def test_type_checking() -> None:
|
|||||||
assert order.price == 7.0
|
assert order.price == 7.0
|
||||||
|
|
||||||
|
|
||||||
|
def test_cached_deserialization() -> None:
|
||||||
|
material = Base(color="blue", opacity=0.5)
|
||||||
|
|
||||||
|
a = Base(name="a")
|
||||||
|
a["@material"] = material
|
||||||
|
b = Base(name="b")
|
||||||
|
b["@material"] = material
|
||||||
|
|
||||||
|
root = Base(a=a, b=b)
|
||||||
|
|
||||||
|
serialized = operations.serialize(root)
|
||||||
|
deserialized = operations.deserialize(serialized)
|
||||||
|
|
||||||
|
assert deserialized["a"]["@material"] is deserialized["b"]["@material"]
|
||||||
|
|||||||
@@ -58,7 +58,7 @@ class TestBranch:
|
|||||||
assert isinstance(branches, list)
|
assert isinstance(branches, list)
|
||||||
assert len(branches) == 2
|
assert len(branches) == 2
|
||||||
assert isinstance(branches[0], Branch)
|
assert isinstance(branches[0], Branch)
|
||||||
assert branches[0].name == branch.name
|
assert branches[1].name == branch.name
|
||||||
|
|
||||||
def test_branch_update(self, client, stream, branch, updated_branch):
|
def test_branch_update(self, client, stream, branch, updated_branch):
|
||||||
updated = client.branch.update(
|
updated = client.branch.update(
|
||||||
|
|||||||
@@ -0,0 +1,48 @@
|
|||||||
|
import pytest
|
||||||
|
from specklepy.api import operations
|
||||||
|
from specklepy.api.client import SpeckleClient
|
||||||
|
from specklepy.objects.base import Base
|
||||||
|
from specklepy.transports.server import ServerTransport
|
||||||
|
from specklepy.api.credentials import Account, get_account_from_token
|
||||||
|
from specklepy.logging.exceptions import SpeckleException, SpeckleWarning
|
||||||
|
|
||||||
|
|
||||||
|
def test_invalid_authentication():
|
||||||
|
client = SpeckleClient()
|
||||||
|
|
||||||
|
with pytest.warns(SpeckleWarning):
|
||||||
|
client.authenticate_with_token("fake token")
|
||||||
|
|
||||||
|
|
||||||
|
def test_invalid_send():
|
||||||
|
client = SpeckleClient()
|
||||||
|
client.account = Account(token="fake_token")
|
||||||
|
transport = ServerTransport("3073b96e86", client)
|
||||||
|
|
||||||
|
with pytest.raises(SpeckleException):
|
||||||
|
operations.send(Base(), [transport])
|
||||||
|
|
||||||
|
|
||||||
|
def test_invalid_receive():
|
||||||
|
client = SpeckleClient()
|
||||||
|
client.account = Account(token="fake_token")
|
||||||
|
transport = ServerTransport("fake stream", client)
|
||||||
|
|
||||||
|
with pytest.raises(SpeckleException):
|
||||||
|
operations.receive("fake object", transport)
|
||||||
|
|
||||||
|
|
||||||
|
def test_account_from_token():
|
||||||
|
token = "fake token"
|
||||||
|
acct = get_account_from_token(token)
|
||||||
|
|
||||||
|
assert acct.token == token
|
||||||
|
|
||||||
|
|
||||||
|
def test_account_from_token_and_url():
|
||||||
|
token = "fake token"
|
||||||
|
url = "fake.server"
|
||||||
|
acct = get_account_from_token(token, url)
|
||||||
|
|
||||||
|
assert acct.token == token
|
||||||
|
assert acct.serverInfo.url == url
|
||||||
@@ -68,3 +68,20 @@ class TestCommit:
|
|||||||
deleted = client.commit.delete(stream_id=stream.id, commit_id=commit_id)
|
deleted = client.commit.delete(stream_id=stream.id, commit_id=commit_id)
|
||||||
|
|
||||||
assert deleted is True
|
assert deleted is True
|
||||||
|
|
||||||
|
def test_commit_marked_as_received(self, client, stream, mesh) -> None:
|
||||||
|
commit = Commit(message="this commit should be received")
|
||||||
|
commit.id = client.commit.create(
|
||||||
|
stream_id=stream.id,
|
||||||
|
object_id=mesh.id,
|
||||||
|
message=commit.message,
|
||||||
|
)
|
||||||
|
|
||||||
|
commit_marked_received = client.commit.received(
|
||||||
|
stream.id,
|
||||||
|
commit.id,
|
||||||
|
source_application="pytest",
|
||||||
|
message="testing received",
|
||||||
|
)
|
||||||
|
|
||||||
|
assert commit_marked_received == True
|
||||||
|
|||||||
+226
-93
@@ -1,15 +1,34 @@
|
|||||||
|
# pylint: disable=redefined-outer-name
|
||||||
import json
|
import json
|
||||||
from typing import Callable
|
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
from specklepy.api import operations
|
from specklepy.api import operations
|
||||||
|
from specklepy.logging.exceptions import SpeckleException
|
||||||
from specklepy.objects.base import Base
|
from specklepy.objects.base import Base
|
||||||
from specklepy.objects.encoding import CurveArray, ObjectArray
|
from specklepy.objects.encoding import CurveArray, ObjectArray
|
||||||
from specklepy.objects.geometry import (Arc, Box, Brep, BrepEdge, BrepFace,
|
from specklepy.objects.geometry import (
|
||||||
BrepLoop, BrepTrim, BrepTrimTypeEnum,
|
Arc,
|
||||||
Circle, Curve, Ellipse, Interval, Line,
|
Box,
|
||||||
Mesh, Plane, Point, Polycurve,
|
Brep,
|
||||||
Polyline, Surface, Vector)
|
BrepEdge,
|
||||||
|
BrepFace,
|
||||||
|
BrepLoop,
|
||||||
|
BrepLoopType,
|
||||||
|
BrepTrim,
|
||||||
|
BrepTrimType,
|
||||||
|
Circle,
|
||||||
|
Curve,
|
||||||
|
Ellipse,
|
||||||
|
Interval,
|
||||||
|
Line,
|
||||||
|
Mesh,
|
||||||
|
Plane,
|
||||||
|
Point,
|
||||||
|
Polycurve,
|
||||||
|
Polyline,
|
||||||
|
Surface,
|
||||||
|
Vector,
|
||||||
|
)
|
||||||
from specklepy.transports.memory import MemoryTransport
|
from specklepy.transports.memory import MemoryTransport
|
||||||
|
|
||||||
|
|
||||||
@@ -30,12 +49,7 @@ def vector():
|
|||||||
|
|
||||||
@pytest.fixture()
|
@pytest.fixture()
|
||||||
def plane(point, vector):
|
def plane(point, vector):
|
||||||
return Plane(
|
return Plane(origin=point, normal=vector, xdir=vector, ydir=vector, units="m")
|
||||||
origin=point,
|
|
||||||
normal=vector,
|
|
||||||
xdir=vector,
|
|
||||||
ydir=vector,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture()
|
@pytest.fixture()
|
||||||
@@ -56,6 +70,7 @@ def line(point, interval):
|
|||||||
start=point,
|
start=point,
|
||||||
end=point,
|
end=point,
|
||||||
domain=interval,
|
domain=interval,
|
||||||
|
units="none"
|
||||||
# These attributes are not handled in C#
|
# These attributes are not handled in C#
|
||||||
# bbox=None,
|
# bbox=None,
|
||||||
# length=None
|
# length=None
|
||||||
@@ -63,7 +78,7 @@ def line(point, interval):
|
|||||||
|
|
||||||
|
|
||||||
@pytest.fixture()
|
@pytest.fixture()
|
||||||
def arc(plane, interval):
|
def arc(plane, interval, point):
|
||||||
return Arc(
|
return Arc(
|
||||||
radius=2.3,
|
radius=2.3,
|
||||||
startAngle=22.1,
|
startAngle=22.1,
|
||||||
@@ -71,14 +86,14 @@ def arc(plane, interval):
|
|||||||
angleRadians=33,
|
angleRadians=33,
|
||||||
plane=plane,
|
plane=plane,
|
||||||
domain=interval,
|
domain=interval,
|
||||||
units='m',
|
units="m",
|
||||||
|
startPoint=point,
|
||||||
|
midPoint=point,
|
||||||
|
endPoint=point,
|
||||||
# These attributes are not handled in C#
|
# These attributes are not handled in C#
|
||||||
# bbox=None,
|
# bbox=None,
|
||||||
# area=None,
|
# area=None,
|
||||||
# length=None,
|
# length=None,
|
||||||
# startPoint=None,
|
|
||||||
# midPoint=None,
|
|
||||||
# endPoint=None,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
@@ -88,7 +103,7 @@ def circle(plane, interval):
|
|||||||
radius=22,
|
radius=22,
|
||||||
plane=plane,
|
plane=plane,
|
||||||
domain=interval,
|
domain=interval,
|
||||||
units='m',
|
units="m",
|
||||||
# These attributes are not handled in C#
|
# These attributes are not handled in C#
|
||||||
# bbox=None,
|
# bbox=None,
|
||||||
# area=None,
|
# area=None,
|
||||||
@@ -103,7 +118,7 @@ def ellipse(plane, interval):
|
|||||||
secondRadius=22,
|
secondRadius=22,
|
||||||
plane=plane,
|
plane=plane,
|
||||||
domain=interval,
|
domain=interval,
|
||||||
units='m',
|
units="m",
|
||||||
# These attributes are not handled in C#
|
# These attributes are not handled in C#
|
||||||
# trimDomain=None,
|
# trimDomain=None,
|
||||||
# bbox=None,
|
# bbox=None,
|
||||||
@@ -118,7 +133,7 @@ def polyline(interval):
|
|||||||
value=[22, 44, 54.3, 99, 232, 21],
|
value=[22, 44, 54.3, 99, 232, 21],
|
||||||
closed=True,
|
closed=True,
|
||||||
domain=interval,
|
domain=interval,
|
||||||
units='m',
|
units="m",
|
||||||
# These attributes are not handled in C#
|
# These attributes are not handled in C#
|
||||||
# bbox=None,
|
# bbox=None,
|
||||||
# area=None,
|
# area=None,
|
||||||
@@ -137,7 +152,7 @@ def curve(interval):
|
|||||||
points=[23, 21, 44, 43, 56, 76, 1, 3, 2],
|
points=[23, 21, 44, 43, 56, 76, 1, 3, 2],
|
||||||
weights=[23, 11, 23],
|
weights=[23, 11, 23],
|
||||||
knots=[22, 45, 76, 11],
|
knots=[22, 45, 76, 11],
|
||||||
units='m',
|
units="m",
|
||||||
# These attributes are not handled in C#
|
# These attributes are not handled in C#
|
||||||
# displayValue=None,
|
# displayValue=None,
|
||||||
# bbox=None,
|
# bbox=None,
|
||||||
@@ -152,7 +167,7 @@ def polycurve(interval, curve, polyline):
|
|||||||
segments=[curve, polyline],
|
segments=[curve, polyline],
|
||||||
domain=interval,
|
domain=interval,
|
||||||
closed=True,
|
closed=True,
|
||||||
units='m',
|
units="m",
|
||||||
# These attributes are not handled in C#
|
# These attributes are not handled in C#
|
||||||
# bbox=None,
|
# bbox=None,
|
||||||
# area=None,
|
# area=None,
|
||||||
@@ -187,7 +202,7 @@ def surface(interval):
|
|||||||
domainV=interval,
|
domainV=interval,
|
||||||
knotsU=[1.1, 2.2, 3.3, 4.4],
|
knotsU=[1.1, 2.2, 3.3, 4.4],
|
||||||
knotsV=[9, 8, 7, 6, 5, 4.4],
|
knotsV=[9, 8, 7, 6, 5, 4.4],
|
||||||
units='m',
|
units="m",
|
||||||
# These attributes are not handled in C#
|
# These attributes are not handled in C#
|
||||||
# bbox=None,
|
# bbox=None,
|
||||||
# area=None,
|
# area=None,
|
||||||
@@ -218,11 +233,7 @@ def brep_edge(interval):
|
|||||||
|
|
||||||
@pytest.fixture()
|
@pytest.fixture()
|
||||||
def brep_loop():
|
def brep_loop():
|
||||||
return BrepLoop(
|
return BrepLoop(FaceIndex=5, TrimIndices=[3, 4, 5], Type=BrepLoopType.Unknown)
|
||||||
FaceIndex=5,
|
|
||||||
TrimIndices=[3, 4, 5],
|
|
||||||
Type='unknown'
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture()
|
@pytest.fixture()
|
||||||
@@ -235,7 +246,7 @@ def brep_trim():
|
|||||||
LoopIndex=4,
|
LoopIndex=4,
|
||||||
CurveIndex=7,
|
CurveIndex=7,
|
||||||
IsoStatus=6,
|
IsoStatus=6,
|
||||||
TrimType='Mated',
|
TrimType=BrepTrimType.Mated,
|
||||||
IsReversed=False,
|
IsReversed=False,
|
||||||
# These attributes are not handled in C#
|
# These attributes are not handled in C#
|
||||||
# Domain=None,
|
# Domain=None,
|
||||||
@@ -243,10 +254,21 @@ def brep_trim():
|
|||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def brep(mesh, box, surface, curve, polyline, circle, point,
|
def brep(
|
||||||
brep_edge, brep_loop, brep_trim, brep_face):
|
mesh,
|
||||||
|
box,
|
||||||
|
surface,
|
||||||
|
curve,
|
||||||
|
polyline,
|
||||||
|
circle,
|
||||||
|
point,
|
||||||
|
brep_edge,
|
||||||
|
brep_loop,
|
||||||
|
brep_trim,
|
||||||
|
brep_face,
|
||||||
|
):
|
||||||
return Brep(
|
return Brep(
|
||||||
provenance='pytest',
|
provenance="pytest",
|
||||||
bbox=box,
|
bbox=box,
|
||||||
area=32,
|
area=32,
|
||||||
volume=54,
|
volume=54,
|
||||||
@@ -265,49 +287,72 @@ def brep(mesh, box, surface, curve, polyline, circle, point,
|
|||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def geometry_objects_dict(point, vector, plane, line, arc,
|
def geometry_objects_dict(
|
||||||
circle, ellipse, polyline, curve,
|
point,
|
||||||
polycurve, surface, brep_trim):
|
vector,
|
||||||
|
plane,
|
||||||
|
line,
|
||||||
|
arc,
|
||||||
|
circle,
|
||||||
|
ellipse,
|
||||||
|
polyline,
|
||||||
|
curve,
|
||||||
|
polycurve,
|
||||||
|
surface,
|
||||||
|
brep_trim,
|
||||||
|
):
|
||||||
return {
|
return {
|
||||||
'point': point,
|
"point": point,
|
||||||
'vector': vector,
|
"vector": vector,
|
||||||
'plane': plane,
|
"plane": plane,
|
||||||
'line': line,
|
"line": line,
|
||||||
'arc': arc,
|
"arc": arc,
|
||||||
'circle': circle,
|
"circle": circle,
|
||||||
'ellipse': ellipse,
|
"ellipse": ellipse,
|
||||||
'polyline': polyline,
|
"polyline": polyline,
|
||||||
'curve': curve,
|
"curve": curve,
|
||||||
'polycurve': polycurve,
|
"polycurve": polycurve,
|
||||||
'surface': surface,
|
"surface": surface,
|
||||||
'brep_trim': brep_trim
|
"brep_trim": brep_trim,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize('object_name', [
|
@pytest.mark.parametrize(
|
||||||
'point', 'vector', 'plane', 'line', 'arc', 'circle',
|
"object_name",
|
||||||
'ellipse', 'polyline', 'curve', 'polycurve', 'surface', 'brep_trim'
|
[
|
||||||
])
|
"point",
|
||||||
|
"vector",
|
||||||
|
"plane",
|
||||||
|
"line",
|
||||||
|
"arc",
|
||||||
|
"circle",
|
||||||
|
"ellipse",
|
||||||
|
"polyline",
|
||||||
|
"curve",
|
||||||
|
"polycurve",
|
||||||
|
"surface",
|
||||||
|
"brep_trim",
|
||||||
|
],
|
||||||
|
)
|
||||||
def test_to_and_from_list(object_name: str, geometry_objects_dict):
|
def test_to_and_from_list(object_name: str, geometry_objects_dict):
|
||||||
object = geometry_objects_dict[object_name]
|
obj = geometry_objects_dict[object_name]
|
||||||
assert hasattr(object, 'to_list')
|
assert hasattr(obj, "to_list")
|
||||||
assert hasattr(object, 'from_list')
|
assert hasattr(obj, "from_list")
|
||||||
|
|
||||||
chunks = object.to_list()
|
chunks = obj.to_list()
|
||||||
assert isinstance(chunks, list)
|
assert isinstance(chunks, list)
|
||||||
|
|
||||||
object_class = object.__class__
|
object_class = obj.__class__
|
||||||
decoded_object: Base = object_class.from_list(chunks)
|
decoded_object: Base = object_class.from_list(chunks)
|
||||||
assert decoded_object.get_id() == object.get_id()
|
assert decoded_object.get_id() == obj.get_id()
|
||||||
|
|
||||||
|
|
||||||
def test_brep_surfaces_value_serialization(surface):
|
def test_brep_surfaces_value_serialization(surface):
|
||||||
brep = Brep()
|
brep = Brep()
|
||||||
assert brep.Surfaces == None
|
assert brep.Surfaces is None
|
||||||
assert brep.SurfacesValue == None
|
assert brep.SurfacesValue is None
|
||||||
brep.Surfaces = [surface, surface]
|
brep.Surfaces = [surface, surface]
|
||||||
assert brep.SurfacesValue == ObjectArray.from_objects(
|
assert brep.SurfacesValue == ObjectArray.from_objects([surface, surface]).data
|
||||||
[surface, surface]).data
|
|
||||||
|
|
||||||
brep.SurfacesValue = ObjectArray.from_objects([surface]).data
|
brep.SurfacesValue = ObjectArray.from_objects([surface]).data
|
||||||
assert len(brep.Surfaces) == 1
|
assert len(brep.Surfaces) == 1
|
||||||
@@ -316,8 +361,8 @@ def test_brep_surfaces_value_serialization(surface):
|
|||||||
|
|
||||||
def test_brep_curve2d_values_serialization(curve, polyline, circle):
|
def test_brep_curve2d_values_serialization(curve, polyline, circle):
|
||||||
brep = Brep()
|
brep = Brep()
|
||||||
assert brep.Curve2D == None
|
assert brep.Curve2D is None
|
||||||
assert brep.Curve2DValues == None
|
assert brep.Curve2DValues is None
|
||||||
brep.Curve2D = [curve, polyline]
|
brep.Curve2D = [curve, polyline]
|
||||||
assert brep.Curve2DValues == CurveArray.from_curves([curve, polyline]).data
|
assert brep.Curve2DValues == CurveArray.from_curves([curve, polyline]).data
|
||||||
|
|
||||||
@@ -328,8 +373,8 @@ def test_brep_curve2d_values_serialization(curve, polyline, circle):
|
|||||||
|
|
||||||
def test_brep_curve3d_values_serialization(curve, polyline, circle):
|
def test_brep_curve3d_values_serialization(curve, polyline, circle):
|
||||||
brep = Brep()
|
brep = Brep()
|
||||||
assert brep.Curve3D == None
|
assert brep.Curve3D is None
|
||||||
assert brep.Curve3DValues == None
|
assert brep.Curve3DValues is None
|
||||||
brep.Curve3D = [curve, polyline]
|
brep.Curve3D = [curve, polyline]
|
||||||
assert brep.Curve3DValues == CurveArray.from_curves([curve, polyline]).data
|
assert brep.Curve3DValues == CurveArray.from_curves([curve, polyline]).data
|
||||||
|
|
||||||
@@ -341,41 +386,107 @@ def test_brep_curve3d_values_serialization(curve, polyline, circle):
|
|||||||
def test_brep_vertices_values_serialization():
|
def test_brep_vertices_values_serialization():
|
||||||
brep = Brep()
|
brep = Brep()
|
||||||
brep.VerticesValue = [1, 1, 1, 1, 2, 2, 2, 3, 3, 3]
|
brep.VerticesValue = [1, 1, 1, 1, 2, 2, 2, 3, 3, 3]
|
||||||
brep.Vertices[0].get_id() == Point(x=1, y=1, z=1, _units='mm').get_id()
|
assert brep.Vertices[0].get_id() == Point(x=1, y=1, z=1, _units="mm").get_id()
|
||||||
brep.Vertices[1].get_id() == Point(x=2, y=2, z=2, _units='mm').get_id()
|
assert brep.Vertices[1].get_id() == Point(x=2, y=2, z=2, _units="mm").get_id()
|
||||||
brep.Vertices[2].get_id() == Point(x=3, y=3, z=3, _units='mm').get_id()
|
assert brep.Vertices[2].get_id() == Point(x=3, y=3, z=3, _units="mm").get_id()
|
||||||
|
|
||||||
|
|
||||||
def test_trims_value_serialization():
|
def test_trims_value_serialization():
|
||||||
brep = Brep()
|
brep = Brep()
|
||||||
brep.TrimsValue = [
|
brep.TrimsValue = [
|
||||||
0, 0, 0, 0, 0, 0, 1, 1, 1,
|
0,
|
||||||
1, 0, 0, 0, 0, 1, 2, 1, 0,
|
0,
|
||||||
|
0,
|
||||||
|
0,
|
||||||
|
0,
|
||||||
|
0,
|
||||||
|
1,
|
||||||
|
1,
|
||||||
|
0,
|
||||||
|
1,
|
||||||
|
0,
|
||||||
|
0,
|
||||||
|
0,
|
||||||
|
0,
|
||||||
|
1,
|
||||||
|
2,
|
||||||
|
1,
|
||||||
|
1,
|
||||||
]
|
]
|
||||||
|
|
||||||
brep.Trims[0].get_id() == BrepTrim(
|
assert (
|
||||||
EdgeIndex=0,
|
brep.Trims[0].get_id()
|
||||||
StartIndex=0,
|
== BrepTrim(
|
||||||
EndIndex=0,
|
EdgeIndex=0,
|
||||||
FaceIndex=0,
|
StartIndex=0,
|
||||||
LoopIndex=0,
|
EndIndex=0,
|
||||||
CurveIndex=0,
|
FaceIndex=0,
|
||||||
IsoStatus=1,
|
LoopIndex=0,
|
||||||
TrimType=BrepTrimTypeEnum.Boundary,
|
CurveIndex=0,
|
||||||
IsReversed=False,
|
IsoStatus=1,
|
||||||
).get_id()
|
TrimType=BrepTrimType.Boundary,
|
||||||
|
IsReversed=False,
|
||||||
|
).get_id()
|
||||||
|
)
|
||||||
|
|
||||||
brep.Trims[1].get_id() == BrepTrim(
|
assert (
|
||||||
EdgeIndex=1,
|
brep.Trims[1].get_id()
|
||||||
StartIndex=0,
|
== BrepTrim(
|
||||||
EndIndex=0,
|
EdgeIndex=1,
|
||||||
FaceIndex=0,
|
StartIndex=0,
|
||||||
LoopIndex=0,
|
EndIndex=0,
|
||||||
CurveIndex=1,
|
FaceIndex=0,
|
||||||
IsoStatus=2,
|
LoopIndex=0,
|
||||||
TrimType=BrepTrimTypeEnum.Boundary,
|
CurveIndex=1,
|
||||||
IsReversed=True,
|
IsoStatus=2,
|
||||||
).get_id()
|
TrimType=BrepTrimType.Boundary,
|
||||||
|
IsReversed=True,
|
||||||
|
).get_id()
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def test_loops_value_serialization():
|
||||||
|
brep = Brep()
|
||||||
|
brep.LoopsValue = [6, 0, 1, 0, 1, 2, 3]
|
||||||
|
|
||||||
|
assert brep == brep.Loops[0]._Brep # pylint: disable=protected-access
|
||||||
|
assert (
|
||||||
|
brep.Loops[0].get_id()
|
||||||
|
== BrepLoop(
|
||||||
|
FaceIndex=0, Type=BrepLoopType(1), TrimIndices=[0, 1, 2, 3]
|
||||||
|
).get_id()
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def test_edges_value_serialization():
|
||||||
|
brep = Brep()
|
||||||
|
brep.EdgesValue = [8, 0, 0, 1, 0, -8.13345756858629, 8.13345756858629, 1, 3]
|
||||||
|
|
||||||
|
assert brep == brep.Edges[0]._Brep # pylint: disable=protected-access
|
||||||
|
assert (
|
||||||
|
brep.Edges[0].get_id()
|
||||||
|
== BrepEdge(
|
||||||
|
Curve3dIndex=0,
|
||||||
|
StartIndex=0,
|
||||||
|
EndIndex=1,
|
||||||
|
ProxyCurveIsReversed=False,
|
||||||
|
Domain=Interval(start=-8.13345756858629, end=8.13345756858629),
|
||||||
|
TrimIndices=[1, 3],
|
||||||
|
).get_id()
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def test_faces_value_serialization():
|
||||||
|
brep = Brep()
|
||||||
|
brep.FacesValue = [4, 0, 0, 1, 0]
|
||||||
|
|
||||||
|
assert brep == brep.Faces[0]._Brep # pylint: disable=protected-access
|
||||||
|
assert (
|
||||||
|
brep.Faces[0].get_id()
|
||||||
|
== BrepFace(
|
||||||
|
SurfaceIndex=0, OuterLoopIndex=0, OrientationReversed=True, LoopIndices=[0]
|
||||||
|
).get_id()
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
def test_serialized_brep_attributes(brep: Brep):
|
def test_serialized_brep_attributes(brep: Brep):
|
||||||
@@ -383,7 +494,29 @@ def test_serialized_brep_attributes(brep: Brep):
|
|||||||
serialized = operations.serialize(brep, [transport])
|
serialized = operations.serialize(brep, [transport])
|
||||||
serialized_dict = json.loads(serialized)
|
serialized_dict = json.loads(serialized)
|
||||||
|
|
||||||
removed_keys = ['Surfaces', 'Curve3D', 'Curve2D', 'Vertices', 'Trims']
|
removed_keys = [
|
||||||
|
"Surfaces",
|
||||||
|
"Curve3D",
|
||||||
|
"Curve2D",
|
||||||
|
"Vertices",
|
||||||
|
"Trims",
|
||||||
|
"Loops",
|
||||||
|
"Edges",
|
||||||
|
"Faces",
|
||||||
|
]
|
||||||
|
|
||||||
for k in removed_keys:
|
for k in removed_keys:
|
||||||
assert k not in serialized_dict.keys()
|
assert k not in serialized_dict.keys()
|
||||||
|
|
||||||
|
|
||||||
|
def test_mesh_create():
|
||||||
|
vertices = [2, 1, 2, 4, 77.3, 5, 33, 4, 2]
|
||||||
|
faces = [1, 2, 3, 4, 5, 6, 7]
|
||||||
|
mesh = Mesh.create(vertices, faces)
|
||||||
|
|
||||||
|
with pytest.raises(SpeckleException):
|
||||||
|
# pylint: disable=unused-variable
|
||||||
|
bad_mesh = Mesh.create(vertices=7, faces=faces)
|
||||||
|
|
||||||
|
assert mesh.vertices == vertices
|
||||||
|
assert mesh.textureCoordinates == []
|
||||||
|
|||||||
+3
-11
@@ -21,11 +21,9 @@ class TestObject:
|
|||||||
|
|
||||||
def test_object_create(self, client, stream, base):
|
def test_object_create(self, client, stream, base):
|
||||||
transport = SQLiteTransport()
|
transport = SQLiteTransport()
|
||||||
s = BaseObjectSerializer(
|
s = BaseObjectSerializer(write_transports=[transport], read_transport=transport)
|
||||||
write_transports=[transport], read_transport=transport)
|
|
||||||
_, base_dict = s.traverse_base(base)
|
_, base_dict = s.traverse_base(base)
|
||||||
obj_id = client.object.create(
|
obj_id = client.object.create(stream_id=stream.id, objects=[base_dict])[0]
|
||||||
stream_id=stream.id, objects=[base_dict])[0]
|
|
||||||
|
|
||||||
assert isinstance(obj_id, str)
|
assert isinstance(obj_id, str)
|
||||||
assert base_dict["@detach"]["speckle_type"] == "reference"
|
assert base_dict["@detach"]["speckle_type"] == "reference"
|
||||||
@@ -43,12 +41,6 @@ class TestObject:
|
|||||||
|
|
||||||
def test_object_array_decoder(self):
|
def test_object_array_decoder(self):
|
||||||
array = ObjectArray()
|
array = ObjectArray()
|
||||||
array.data = [
|
array.data = [5, 1, 1, 1, 1, 1, 4, 1, 1, 1, 1, 3, 1, 1, 1, 2, 1, 1, 1, 1]
|
||||||
5, 1, 1, 1, 1, 1,
|
|
||||||
4, 1, 1, 1, 1,
|
|
||||||
3, 1, 1, 1,
|
|
||||||
2, 1, 1,
|
|
||||||
1, 1
|
|
||||||
]
|
|
||||||
|
|
||||||
assert array.decode(decoder=sum) == [5, 4, 3, 2, 1]
|
assert array.decode(decoder=sum) == [5, 4, 3, 2, 1]
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
import json
|
import json
|
||||||
from attr import has
|
|
||||||
import pytest
|
import pytest
|
||||||
from specklepy.api import operations
|
from specklepy.api import operations
|
||||||
from specklepy.transports.server import ServerTransport
|
from specklepy.transports.server import ServerTransport
|
||||||
@@ -53,6 +52,11 @@ class TestSerialization:
|
|||||||
def test_send_and_receive(self, client, sample_stream, mesh):
|
def test_send_and_receive(self, client, sample_stream, mesh):
|
||||||
transport = ServerTransport(client=client, stream_id=sample_stream.id)
|
transport = ServerTransport(client=client, stream_id=sample_stream.id)
|
||||||
hash = operations.send(mesh, transports=[transport])
|
hash = operations.send(mesh, transports=[transport])
|
||||||
|
|
||||||
|
# also try constructing server transport with token and url
|
||||||
|
transport = ServerTransport(
|
||||||
|
stream_id=sample_stream.id, token=client.account.token, url=client.url
|
||||||
|
)
|
||||||
# use a fresh memory transport to force receiving from remote
|
# use a fresh memory transport to force receiving from remote
|
||||||
received = operations.receive(
|
received = operations.receive(
|
||||||
hash, remote_transport=transport, local_transport=MemoryTransport()
|
hash, remote_transport=transport, local_transport=MemoryTransport()
|
||||||
@@ -84,4 +88,10 @@ class TestSerialization:
|
|||||||
untyped = '{"foo": "bar"}'
|
untyped = '{"foo": "bar"}'
|
||||||
deserialised = operations.deserialize(untyped)
|
deserialised = operations.deserialize(untyped)
|
||||||
|
|
||||||
assert deserialised == {"foo": "bar"}
|
assert deserialised == {"foo": "bar"}
|
||||||
|
|
||||||
|
def test_big_int(self):
|
||||||
|
big_int = '{"big": ' + str(2 ** 64) + "}"
|
||||||
|
deserialised = operations.deserialize(big_int)
|
||||||
|
|
||||||
|
assert deserialised == {"big": 2 ** 64}
|
||||||
|
|||||||
+10
-2
@@ -1,5 +1,6 @@
|
|||||||
import pytest
|
import pytest
|
||||||
from specklepy.api.models import ServerInfo
|
from specklepy.api.models import ServerInfo
|
||||||
|
from specklepy.api.client import SpeckleClient
|
||||||
|
|
||||||
|
|
||||||
class TestServer:
|
class TestServer:
|
||||||
@@ -12,12 +13,19 @@ class TestServer:
|
|||||||
"lifespan": 9001,
|
"lifespan": 9001,
|
||||||
}
|
}
|
||||||
|
|
||||||
def test_server_get(self, client):
|
def test_server_get(self, client: SpeckleClient):
|
||||||
server = client.server.get()
|
server = client.server.get()
|
||||||
|
|
||||||
assert isinstance(server, ServerInfo)
|
assert isinstance(server, ServerInfo)
|
||||||
|
|
||||||
def test_server_apps(self, client):
|
def test_server_version(self, client: SpeckleClient):
|
||||||
|
version = client.server.version()
|
||||||
|
|
||||||
|
assert isinstance(version, tuple)
|
||||||
|
assert isinstance(version[0], int)
|
||||||
|
assert len(version) >= 3
|
||||||
|
|
||||||
|
def test_server_apps(self, client: SpeckleClient):
|
||||||
apps = client.server.apps()
|
apps = client.server.apps()
|
||||||
|
|
||||||
assert isinstance(apps, list)
|
assert isinstance(apps, list)
|
||||||
|
|||||||
+143
-13
@@ -1,6 +1,18 @@
|
|||||||
import pytest
|
import pytest
|
||||||
from specklepy.api.models import Stream
|
from datetime import datetime
|
||||||
from specklepy.logging.exceptions import GraphQLException
|
from specklepy.api.models import (
|
||||||
|
ActivityCollection,
|
||||||
|
Activity,
|
||||||
|
PendingStreamCollaborator,
|
||||||
|
Stream,
|
||||||
|
User,
|
||||||
|
)
|
||||||
|
from specklepy.api.client import SpeckleClient
|
||||||
|
from specklepy.logging.exceptions import (
|
||||||
|
GraphQLException,
|
||||||
|
SpeckleException,
|
||||||
|
UnsupportedException,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.run(order=2)
|
@pytest.mark.run(order=2)
|
||||||
@@ -23,6 +35,10 @@ class TestStream:
|
|||||||
isPublic=False,
|
isPublic=False,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@pytest.fixture(scope="module")
|
||||||
|
def second_user(self, second_client: SpeckleClient):
|
||||||
|
return second_client.user.get()
|
||||||
|
|
||||||
def test_stream_create(self, client, stream, updated_stream):
|
def test_stream_create(self, client, stream, updated_stream):
|
||||||
stream.id = updated_stream.id = client.stream.create(
|
stream.id = updated_stream.id = client.stream.create(
|
||||||
name=stream.name,
|
name=stream.name,
|
||||||
@@ -67,29 +83,143 @@ class TestStream:
|
|||||||
assert len(search_results) == 1
|
assert len(search_results) == 1
|
||||||
assert search_results[0].name == updated_stream.name
|
assert search_results[0].name == updated_stream.name
|
||||||
|
|
||||||
def test_stream_grant_permission(self, client, stream, second_user_dict):
|
def test_stream_favorite(self, client, stream):
|
||||||
granted = client.stream.grant_permission(
|
favorited = client.stream.favorite(stream.id)
|
||||||
|
|
||||||
|
assert isinstance(favorited, Stream)
|
||||||
|
assert favorited.favoritedDate is not None
|
||||||
|
|
||||||
|
unfavorited = client.stream.favorite(stream.id, False)
|
||||||
|
assert isinstance(favorited, Stream)
|
||||||
|
assert unfavorited.favoritedDate is None
|
||||||
|
|
||||||
|
def test_stream_grant_permission(self, client, stream, second_user):
|
||||||
|
# deprecated as of Speckle Server 2.6.4
|
||||||
|
with pytest.raises(UnsupportedException):
|
||||||
|
client.stream.grant_permission(
|
||||||
|
stream_id=stream.id,
|
||||||
|
user_id=second_user.id,
|
||||||
|
role="stream:contributor",
|
||||||
|
)
|
||||||
|
|
||||||
|
def test_stream_invite(
|
||||||
|
self, client: SpeckleClient, stream: Stream, second_user_dict: dict
|
||||||
|
):
|
||||||
|
invited = client.stream.invite(
|
||||||
stream_id=stream.id,
|
stream_id=stream.id,
|
||||||
user_id=second_user_dict["id"],
|
email=second_user_dict["email"],
|
||||||
role="stream:contributor",
|
role="stream:reviewer",
|
||||||
|
message="welcome to my stream!",
|
||||||
)
|
)
|
||||||
|
|
||||||
fetched_stream = client.stream.get(stream.id)
|
assert invited is True
|
||||||
|
|
||||||
assert granted is True
|
# fail if no email or id
|
||||||
assert len(fetched_stream.collaborators) == 2
|
with pytest.raises(SpeckleException):
|
||||||
assert fetched_stream.collaborators[0].name == second_user_dict["name"]
|
client.stream.invite(stream_id=stream.id)
|
||||||
|
|
||||||
def test_stream_revoke_permission(self, client, stream, second_user_dict):
|
def test_stream_invite_get_all_for_user(
|
||||||
|
self, second_client: SpeckleClient, stream: Stream
|
||||||
|
):
|
||||||
|
# NOTE: these are user queries, but testing here to contain the flow
|
||||||
|
invites = second_client.user.get_all_pending_invites()
|
||||||
|
|
||||||
|
assert isinstance(invites, list)
|
||||||
|
assert isinstance(invites[0], PendingStreamCollaborator)
|
||||||
|
assert len(invites) == 1
|
||||||
|
|
||||||
|
invite = second_client.user.get_pending_invite(stream_id=stream.id)
|
||||||
|
assert isinstance(invite, PendingStreamCollaborator)
|
||||||
|
|
||||||
|
def test_stream_invite_use(self, second_client: SpeckleClient, stream: Stream):
|
||||||
|
invite: PendingStreamCollaborator = (
|
||||||
|
second_client.user.get_all_pending_invites()[0]
|
||||||
|
)
|
||||||
|
|
||||||
|
accepted = second_client.stream.invite_use(
|
||||||
|
stream_id=stream.id, token=invite.token
|
||||||
|
)
|
||||||
|
|
||||||
|
assert accepted is True
|
||||||
|
|
||||||
|
def test_stream_update_permission(
|
||||||
|
self, client: SpeckleClient, stream: Stream, second_user: User
|
||||||
|
):
|
||||||
|
updated = client.stream.update_permission(
|
||||||
|
stream_id=stream.id, user_id=second_user.id, role="stream:contributor"
|
||||||
|
)
|
||||||
|
|
||||||
|
assert updated is True
|
||||||
|
|
||||||
|
def test_stream_revoke_permission(self, client, stream, second_user):
|
||||||
revoked = client.stream.revoke_permission(
|
revoked = client.stream.revoke_permission(
|
||||||
stream_id=stream.id, user_id=second_user_dict["id"]
|
stream_id=stream.id, user_id=second_user.id
|
||||||
)
|
)
|
||||||
|
|
||||||
fetched_stream = client.stream.get(stream.id)
|
fetched_stream = client.stream.get(stream.id)
|
||||||
|
|
||||||
assert revoked == True
|
assert revoked is True
|
||||||
assert len(fetched_stream.collaborators) == 1
|
assert len(fetched_stream.collaborators) == 1
|
||||||
|
|
||||||
|
def test_stream_invite_cancel(
|
||||||
|
self,
|
||||||
|
client: SpeckleClient,
|
||||||
|
stream: Stream,
|
||||||
|
second_user: User,
|
||||||
|
):
|
||||||
|
invited = client.stream.invite(
|
||||||
|
stream_id=stream.id,
|
||||||
|
user_id=second_user.id,
|
||||||
|
message="welcome to my stream!",
|
||||||
|
)
|
||||||
|
assert invited is True
|
||||||
|
|
||||||
|
invites = client.stream.get_all_pending_invites(stream_id=stream.id)
|
||||||
|
|
||||||
|
cancelled = client.stream.invite_cancel(
|
||||||
|
invite_id=invites[0].inviteId, stream_id=stream.id
|
||||||
|
)
|
||||||
|
|
||||||
|
assert cancelled is True
|
||||||
|
|
||||||
|
def test_stream_invite_batch(
|
||||||
|
self, client: SpeckleClient, stream: Stream, second_user: User
|
||||||
|
):
|
||||||
|
# NOTE: only works for server admins
|
||||||
|
# invited = client.stream.invite_batch(
|
||||||
|
# stream_id=stream.id,
|
||||||
|
# emails=["userA@speckle.xyz", "userB@speckle.xyz"],
|
||||||
|
# user_ids=[second_user.id],
|
||||||
|
# message="yeehaw 🤠",
|
||||||
|
# )
|
||||||
|
|
||||||
|
# assert invited is True
|
||||||
|
|
||||||
|
# invited_only_email = client.stream.invite_batch(
|
||||||
|
# stream_id=stream.id,
|
||||||
|
# emails=["userC@speckle.xyz"],
|
||||||
|
# message="yeehaw 🤠",
|
||||||
|
# )
|
||||||
|
|
||||||
|
# assert invited_only_email is True
|
||||||
|
|
||||||
|
# fail if no emails or user ids
|
||||||
|
with pytest.raises(SpeckleException):
|
||||||
|
client.stream.invite_batch(stream_id=stream.id)
|
||||||
|
|
||||||
|
def test_stream_activity(self, client: SpeckleClient, stream: Stream):
|
||||||
|
activity = client.stream.activity(stream.id)
|
||||||
|
|
||||||
|
older_activity = client.stream.activity(
|
||||||
|
stream.id, before=activity.items[0].time
|
||||||
|
)
|
||||||
|
|
||||||
|
assert isinstance(activity, ActivityCollection)
|
||||||
|
assert isinstance(older_activity, ActivityCollection)
|
||||||
|
assert older_activity.totalCount < activity.totalCount
|
||||||
|
assert activity.items is not None
|
||||||
|
assert isinstance(activity.items[0], Activity)
|
||||||
|
|
||||||
def test_stream_delete(self, client, stream):
|
def test_stream_delete(self, client, stream):
|
||||||
deleted = client.stream.delete(stream.id)
|
deleted = client.stream.delete(stream.id)
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,158 @@
|
|||||||
|
import json
|
||||||
|
from typing import Callable
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
from specklepy.api import operations
|
||||||
|
from specklepy.logging.exceptions import SpeckleException
|
||||||
|
from specklepy.objects.base import Base
|
||||||
|
from specklepy.objects.encoding import CurveArray, ObjectArray
|
||||||
|
from specklepy.objects.geometry import (
|
||||||
|
Line,
|
||||||
|
Mesh,
|
||||||
|
Point,
|
||||||
|
Vector,
|
||||||
|
)
|
||||||
|
from specklepy.transports.memory import MemoryTransport
|
||||||
|
from specklepy.objects.structural.geometry import (
|
||||||
|
Node,
|
||||||
|
Element1D,
|
||||||
|
Element2D,
|
||||||
|
Restraint,
|
||||||
|
ElementType1D,
|
||||||
|
ElementType2D,
|
||||||
|
)
|
||||||
|
from specklepy.objects.structural.properties import (
|
||||||
|
Property1D,
|
||||||
|
Property2D,
|
||||||
|
SectionProfile,
|
||||||
|
MemberType,
|
||||||
|
ShapeType,
|
||||||
|
)
|
||||||
|
from specklepy.objects.structural.material import (
|
||||||
|
Material,
|
||||||
|
)
|
||||||
|
from specklepy.objects.structural.analysis import Model
|
||||||
|
|
||||||
|
from specklepy.objects.structural.loading import LoadGravity
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def point():
|
||||||
|
return Point(x=1, y=10, z=0)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def vector():
|
||||||
|
return Vector(x=0, y=0, z=-1)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def line(point, interval):
|
||||||
|
return Line(
|
||||||
|
start=point,
|
||||||
|
end=point,
|
||||||
|
domain=interval,
|
||||||
|
# These attributes are not handled in C#
|
||||||
|
# bbox=None,
|
||||||
|
# length=None
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def mesh(box):
|
||||||
|
return Mesh(
|
||||||
|
vertices=[2, 1, 2, 4, 77.3, 5, 33, 4, 2],
|
||||||
|
faces=[1, 2, 3, 4, 5, 6, 7],
|
||||||
|
colors=[111, 222, 333, 444, 555, 666, 777],
|
||||||
|
bbox=box,
|
||||||
|
area=233,
|
||||||
|
volume=232.2,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def restraint():
|
||||||
|
return Restraint(code="FFFFFF")
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def node(restraint, point):
|
||||||
|
return Node(basePoint=point, restraint=restraint, name="node1")
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def material():
|
||||||
|
return Material(name="TestMaterial")
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def memberType():
|
||||||
|
return MemberType(0)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def shapeType():
|
||||||
|
return ShapeType(8)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def sectionProfile(shapeType):
|
||||||
|
return SectionProfile(name="Test", shapeType=shapeType)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def property1D(memberType, sectionProfile, material):
|
||||||
|
return Property1D(
|
||||||
|
Material=material,
|
||||||
|
SectionProfile=sectionProfile,
|
||||||
|
memberType=memberType,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def elementType1D():
|
||||||
|
return ElementType1D(0)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def element1D(line, restraint, elementType1D, property1D):
|
||||||
|
return Element1D(
|
||||||
|
baseLine=line,
|
||||||
|
end1Releases=restraint,
|
||||||
|
end2Releases=restraint,
|
||||||
|
type=elementType1D,
|
||||||
|
property=property1D,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def property2D(material):
|
||||||
|
return Property2D(Material=material)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def elementType2D():
|
||||||
|
return ElementType2D(0)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def element2D(point, elementType2D):
|
||||||
|
return Element2D(
|
||||||
|
topology=[point],
|
||||||
|
type=elementType2D,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def loadGravity(element1D, element2D, vector):
|
||||||
|
return LoadGravity(elements=[element1D, element2D], gravityFactors=vector)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def model(loadGravity, element1D, element2D, material, property1D, property2D):
|
||||||
|
return Model(
|
||||||
|
loads=[loadGravity],
|
||||||
|
elements=[element1D, element2D],
|
||||||
|
materials=[material],
|
||||||
|
properties=[property1D, property2D],
|
||||||
|
)
|
||||||
@@ -0,0 +1,132 @@
|
|||||||
|
from typing import List
|
||||||
|
import pytest
|
||||||
|
from specklepy.api import operations
|
||||||
|
from specklepy.objects.geometry import Point, Vector
|
||||||
|
from specklepy.objects.other import (
|
||||||
|
Transform,
|
||||||
|
BlockInstance,
|
||||||
|
BlockDefinition,
|
||||||
|
IDENTITY_TRANSFORM,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def point():
|
||||||
|
return Point(x=1, y=10, z=2)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def points():
|
||||||
|
return [Point(x=1 + i, y=10 + i, z=2 + i) for i in range(5)]
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def point_value():
|
||||||
|
return [1, 10, 2]
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def points_values():
|
||||||
|
coords = []
|
||||||
|
for i in range(5):
|
||||||
|
coords.extend([1 + i, 10 + i, 2 + 1])
|
||||||
|
return coords
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def vector():
|
||||||
|
return Vector(x=1, y=10, z=2)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def vector_value():
|
||||||
|
return [1, 1, 2]
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def transform():
|
||||||
|
"""Translates to [1, 2, 0] and scales z by 0.5"""
|
||||||
|
return Transform.from_list(
|
||||||
|
[
|
||||||
|
1.0,
|
||||||
|
0.0,
|
||||||
|
0.0,
|
||||||
|
1.0,
|
||||||
|
0.0,
|
||||||
|
1.0,
|
||||||
|
0.0,
|
||||||
|
2.0,
|
||||||
|
0.0,
|
||||||
|
0.0,
|
||||||
|
0.5,
|
||||||
|
0.0,
|
||||||
|
0.0,
|
||||||
|
0.0,
|
||||||
|
0.0,
|
||||||
|
1.0,
|
||||||
|
]
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def test_point_transform(point: Point, transform: Transform):
|
||||||
|
new_point = transform.apply_to_point(point)
|
||||||
|
|
||||||
|
assert new_point.x == point.x + 1
|
||||||
|
assert new_point.y == point.y + 2
|
||||||
|
assert new_point.z == point.z * 0.5
|
||||||
|
|
||||||
|
|
||||||
|
def test_points_transform(points: List[Point], transform: Transform):
|
||||||
|
new_points = transform.apply_to_points(points)
|
||||||
|
|
||||||
|
for (i, new_point) in enumerate(new_points):
|
||||||
|
assert new_point.x == points[i].x + 1
|
||||||
|
assert new_point.y == points[i].y + 2
|
||||||
|
assert new_point.z == points[i].z * 0.5
|
||||||
|
|
||||||
|
|
||||||
|
def test_point_value_transform(point_value: List[float], transform: Transform):
|
||||||
|
new_coords = transform.apply_to_point_value(point_value)
|
||||||
|
|
||||||
|
assert new_coords[0] == point_value[0] + 1
|
||||||
|
assert new_coords[1] == point_value[1] + 2
|
||||||
|
assert new_coords[2] == point_value[2] * 0.5
|
||||||
|
|
||||||
|
|
||||||
|
def test_points_values_transform(points_values: List[float], transform: Transform):
|
||||||
|
new_coords = transform.apply_to_points_values(points_values)
|
||||||
|
|
||||||
|
for i in range(0, len(points_values), 3):
|
||||||
|
assert new_coords[i] == points_values[i] + 1
|
||||||
|
assert new_coords[i + 1] == points_values[i + 1] + 2
|
||||||
|
assert new_coords[i + 2] == points_values[i + 2] * 0.5
|
||||||
|
|
||||||
|
|
||||||
|
def test_vector_transform(vector: Vector, transform: Transform):
|
||||||
|
new_vector = transform.apply_to_vector(vector)
|
||||||
|
|
||||||
|
assert new_vector.x == vector.x
|
||||||
|
assert new_vector.y == vector.y
|
||||||
|
assert new_vector.z == vector.z * 0.5
|
||||||
|
|
||||||
|
|
||||||
|
def test_vector_value_transform(vector_value: List[float], transform: Transform):
|
||||||
|
new_coords = transform.apply_to_vector_value(vector_value)
|
||||||
|
|
||||||
|
assert new_coords[0] == vector_value[0]
|
||||||
|
assert new_coords[1] == vector_value[1]
|
||||||
|
assert new_coords[2] == vector_value[2] * 0.5
|
||||||
|
|
||||||
|
|
||||||
|
def test_transform_fails_with_malformed_value():
|
||||||
|
with pytest.raises(ValueError):
|
||||||
|
Transform.from_list("asdf")
|
||||||
|
with pytest.raises(ValueError):
|
||||||
|
Transform.from_list([7, 8, 9])
|
||||||
|
|
||||||
|
|
||||||
|
def test_transform_serialisation(transform: Transform):
|
||||||
|
serialized = operations.serialize(transform)
|
||||||
|
deserialized = operations.deserialize(serialized)
|
||||||
|
|
||||||
|
assert transform.get_id() == deserialized.get_id()
|
||||||
+17
-2
@@ -1,6 +1,7 @@
|
|||||||
from specklepy.logging.exceptions import SpeckleException
|
|
||||||
from specklepy.api.models import User
|
|
||||||
import pytest
|
import pytest
|
||||||
|
from specklepy.api.client import SpeckleClient
|
||||||
|
from specklepy.api.models import Activity, ActivityCollection, User
|
||||||
|
from specklepy.logging.exceptions import SpeckleException
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.run(order=1)
|
@pytest.mark.run(order=1)
|
||||||
@@ -43,3 +44,17 @@ class TestUser:
|
|||||||
assert isinstance(failed_update, SpeckleException)
|
assert isinstance(failed_update, SpeckleException)
|
||||||
assert updated is True
|
assert updated is True
|
||||||
assert me.bio == bio
|
assert me.bio == bio
|
||||||
|
|
||||||
|
def test_user_activity(self, client: SpeckleClient, second_user_dict):
|
||||||
|
my_activity = client.user.activity(limit=10)
|
||||||
|
their_activity = client.user.activity(second_user_dict["id"])
|
||||||
|
|
||||||
|
assert isinstance(my_activity, ActivityCollection)
|
||||||
|
assert isinstance(my_activity.items[0], Activity)
|
||||||
|
assert my_activity.totalCount > 0
|
||||||
|
assert isinstance(their_activity, ActivityCollection)
|
||||||
|
|
||||||
|
older_activity = client.user.activity(before=my_activity.items[0].time)
|
||||||
|
|
||||||
|
assert isinstance(older_activity, ActivityCollection)
|
||||||
|
assert older_activity.totalCount < my_activity.totalCount
|
||||||
|
|||||||
+73
-33
@@ -1,41 +1,81 @@
|
|||||||
import pytest
|
import pytest
|
||||||
from specklepy.api.credentials import StreamWrapper
|
from specklepy.api.wrapper import StreamWrapper
|
||||||
|
|
||||||
|
|
||||||
class TestWrapper:
|
def test_parse_stream():
|
||||||
def test_parse_stream(self):
|
wrap = StreamWrapper("https://testing.speckle.dev/streams/a75ab4f10f")
|
||||||
wrap = StreamWrapper("https://testing.speckle.dev/streams/a75ab4f10f")
|
assert wrap.type == "stream"
|
||||||
assert wrap.type == "stream"
|
|
||||||
|
|
||||||
def test_parse_branch(self):
|
|
||||||
wacky_wrap = StreamWrapper(
|
|
||||||
"https://testing.speckle.dev/streams/4c3ce1459c/branches/%F0%9F%8D%95%E2%AC%85%F0%9F%8C%9F%20you%20wat%3F"
|
|
||||||
)
|
|
||||||
wrap = StreamWrapper(
|
|
||||||
"https://testing.speckle.dev/streams/4c3ce1459c/branches/next%20level"
|
|
||||||
)
|
|
||||||
assert wacky_wrap.type == "branch"
|
|
||||||
assert wacky_wrap.branch_name == "🍕⬅🌟 you wat?"
|
|
||||||
assert wrap.type == "branch"
|
|
||||||
|
|
||||||
def test_parse_commit(self):
|
def test_parse_branch():
|
||||||
wrap = StreamWrapper(
|
wacky_wrap = StreamWrapper(
|
||||||
"https://testing.speckle.dev/streams/4c3ce1459c/commits/8b9b831792"
|
"https://testing.speckle.dev/streams/4c3ce1459c/branches/%F0%9F%8D%95%E2%AC%85%F0%9F%8C%9F%20you%20wat%3F"
|
||||||
)
|
)
|
||||||
assert wrap.type == "commit"
|
wrap = StreamWrapper(
|
||||||
|
"https://testing.speckle.dev/streams/4c3ce1459c/branches/next%20level"
|
||||||
|
)
|
||||||
|
assert wacky_wrap.type == "branch"
|
||||||
|
assert wacky_wrap.branch_name == "🍕⬅🌟 you wat?"
|
||||||
|
assert wrap.type == "branch"
|
||||||
|
|
||||||
def test_parse_object(self):
|
|
||||||
wrap = StreamWrapper(
|
|
||||||
"https://testing.speckle.dev/streams/a75ab4f10f/objects/5530363e6d51c904903dafc3ea1d2ec6"
|
|
||||||
)
|
|
||||||
assert wrap.type == "object"
|
|
||||||
|
|
||||||
def test_parse_globals_as_branch(self):
|
def test_parse_nested_branch():
|
||||||
wrap = StreamWrapper("https://testing.speckle.dev/streams/0c6ad366c4/globals/")
|
wrap = StreamWrapper(
|
||||||
assert wrap.type == "branch"
|
"https://testing.speckle.dev/streams/4c3ce1459c/branches/izzy/dev"
|
||||||
|
)
|
||||||
|
|
||||||
def test_parse_globals_as_commit(self):
|
assert wrap.branch_name == "izzy/dev"
|
||||||
wrap = StreamWrapper(
|
assert wrap.type == "branch"
|
||||||
"https://testing.speckle.dev/streams/0c6ad366c4/globals/abd3787893"
|
|
||||||
)
|
|
||||||
assert wrap.type == "commit"
|
def test_parse_commit():
|
||||||
|
wrap = StreamWrapper(
|
||||||
|
"https://testing.speckle.dev/streams/4c3ce1459c/commits/8b9b831792"
|
||||||
|
)
|
||||||
|
assert wrap.type == "commit"
|
||||||
|
|
||||||
|
|
||||||
|
def test_parse_object():
|
||||||
|
wrap = StreamWrapper(
|
||||||
|
"https://testing.speckle.dev/streams/a75ab4f10f/objects/5530363e6d51c904903dafc3ea1d2ec6"
|
||||||
|
)
|
||||||
|
assert wrap.type == "object"
|
||||||
|
|
||||||
|
|
||||||
|
def test_parse_globals_as_branch():
|
||||||
|
wrap = StreamWrapper("https://testing.speckle.dev/streams/0c6ad366c4/globals/")
|
||||||
|
assert wrap.type == "branch"
|
||||||
|
|
||||||
|
|
||||||
|
def test_parse_globals_as_commit():
|
||||||
|
wrap = StreamWrapper(
|
||||||
|
"https://testing.speckle.dev/streams/0c6ad366c4/globals/abd3787893"
|
||||||
|
)
|
||||||
|
assert wrap.type == "commit"
|
||||||
|
|
||||||
|
|
||||||
|
#! NOTE: the following three tests may not pass locally if you have a `speckle.xyz` account in manager
|
||||||
|
def test_get_client_without_auth():
|
||||||
|
wrap = StreamWrapper("https://speckle.xyz/streams/4c3ce1459c/commits/8b9b831792")
|
||||||
|
client = wrap.get_client()
|
||||||
|
|
||||||
|
assert client is not None
|
||||||
|
|
||||||
|
|
||||||
|
def test_get_new_client_with_token():
|
||||||
|
wrap = StreamWrapper("https://speckle.xyz/streams/4c3ce1459c/commits/8b9b831792")
|
||||||
|
client = wrap.get_client()
|
||||||
|
client = wrap.get_client(token="super-secret-token")
|
||||||
|
|
||||||
|
assert client.account.token == "super-secret-token"
|
||||||
|
|
||||||
|
|
||||||
|
def test_get_transport_with_token():
|
||||||
|
wrap = StreamWrapper("https://speckle.xyz/streams/4c3ce1459c/commits/8b9b831792")
|
||||||
|
client = wrap.get_client()
|
||||||
|
assert not client.account.token # unauthenticated bc no local accounts
|
||||||
|
|
||||||
|
transport = wrap.get_transport(token="super-secret-token")
|
||||||
|
|
||||||
|
assert transport is not None
|
||||||
|
assert client.account.token == "super-secret-token"
|
||||||
|
|||||||
Reference in New Issue
Block a user