diff --git a/speckle/api/credentials.py b/speckle/api/credentials.py index 78b0c25..4f89bac 100644 --- a/speckle/api/credentials.py +++ b/speckle/api/credentials.py @@ -1,5 +1,3 @@ -from __future__ import annotations - from typing import List, Optional from pydantic import BaseModel from speckle.transports.sqlite import SQLiteTransport diff --git a/speckle/api/models.py b/speckle/api/models.py index 43d0fcf..83a08c9 100644 --- a/speckle/api/models.py +++ b/speckle/api/models.py @@ -2,8 +2,6 @@ # filename: stream_schema.json # timestamp: 2020-11-17T14:33:13+00:00 -from __future__ import annotations - from datetime import datetime from typing import Any, Dict, List, Optional @@ -64,12 +62,6 @@ class Branches(BaseModel): items: List[Branch] = [] -class Streams(BaseModel): - totalCount: Optional[int] - cursor: Optional[datetime] - items: List[Stream] = [] - - class Stream(BaseModel): id: Optional[str] name: Optional[str] @@ -89,6 +81,12 @@ class Stream(BaseModel): return self.__repr__() +class Streams(BaseModel): + totalCount: Optional[int] + cursor: Optional[datetime] + items: List[Stream] = [] + + class User(BaseModel): id: Optional[str] email: Optional[str] diff --git a/speckle/objects/base.py b/speckle/objects/base.py index 94d0768..723a4d3 100644 --- a/speckle/objects/base.py +++ b/speckle/objects/base.py @@ -1,5 +1,3 @@ -from __future__ import annotations - from pydantic import BaseModel from pydantic.main import Extra from typing import Dict, List, Optional, Any @@ -100,7 +98,7 @@ class Base(BaseModel): serializer.write_transports = [MemoryTransport()] return serializer.traverse_base(self)[0] - def _count_descendants(self, base: Base, parsed: List) -> int: + def _count_descendants(self, base: "Base", parsed: List) -> int: if base in parsed: return 0 parsed.append(base) @@ -119,20 +117,20 @@ class Base(BaseModel): count = 0 if obj == None: return count - if isinstance(obj, Base): + if isinstance(obj, "Base"): count += 1 count += self._count_descendants(obj, parsed) return count elif isinstance(obj, list): for item in obj: - if isinstance(item, Base): + if isinstance(item, "Base"): count += 1 count += self._count_descendants(item, parsed) else: count += self._handle_object_count(item, parsed) elif isinstance(obj, dict): for _, value in obj.items(): - if isinstance(value, Base): + if isinstance(value, "Base"): count += 1 count += self._count_descendants(value, parsed) else: diff --git a/speckle/objects/fakemesh.py b/speckle/objects/fakemesh.py index 82847ac..991c4d9 100644 --- a/speckle/objects/fakemesh.py +++ b/speckle/objects/fakemesh.py @@ -1,5 +1,3 @@ -from __future__ import annotations - from typing import List, Optional from .base import Base diff --git a/speckle/objects/mesh.py b/speckle/objects/mesh.py index b15ca51..f7375b8 100644 --- a/speckle/objects/mesh.py +++ b/speckle/objects/mesh.py @@ -1,5 +1,3 @@ -from __future__ import annotations - from typing import List, Optional from pydantic import BaseModel