-
Notifications
You must be signed in to change notification settings - Fork 35
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add database infrastructure and mock db
- Loading branch information
Showing
9 changed files
with
302 additions
and
6 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,3 +1,5 @@ | ||
# See https://help.github.com/articles/ignoring-files/ for more about ignoring files. | ||
|
||
frontend/node_modules/ | ||
venv/ | ||
.vscode |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Empty file.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,14 @@ | ||
from sqlalchemy import create_engine | ||
from sqlalchemy.ext.declarative import declarative_base | ||
from sqlalchemy.orm import sessionmaker | ||
|
||
# TODO: Move to config | ||
SQLALCHEMY_DATABASE_URL = "sqlite:///:memory:" | ||
|
||
engine = create_engine( | ||
SQLALCHEMY_DATABASE_URL, connect_args={"check_same_thread": False} | ||
) | ||
|
||
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine) | ||
|
||
Base = declarative_base() |
148 changes: 148 additions & 0 deletions
148
backend/src/flotilla/database/mock_database/mock_database.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,148 @@ | ||
import datetime | ||
import resource | ||
|
||
from flotilla.database.models import ( | ||
Capability, | ||
Entry, | ||
EntryStatus, | ||
Event, | ||
InspectionType, | ||
Report, | ||
ReportStatus, | ||
Resource, | ||
Robot, | ||
RobotStatus, | ||
Topic, | ||
) | ||
|
||
|
||
def populate_mock_db(session, engine, base) -> None: | ||
base.metadata.create_all(engine) | ||
|
||
robot_1 = Robot( | ||
name="Harald", | ||
model="King", | ||
serial_number="V", | ||
logs="", | ||
status=RobotStatus.available, | ||
) | ||
|
||
robot_2 = Robot( | ||
name="Haakon", | ||
model="King", | ||
serial_number="VII", | ||
logs="", | ||
status=RobotStatus.offline, | ||
) | ||
|
||
session.add_all([robot_1, robot_2]) | ||
session.commit() | ||
|
||
report_1 = Report( | ||
robot_id=robot_1.id, | ||
isar_mission_id="isar_mission_id", | ||
echo_mission_id=1, | ||
log="", | ||
status=ReportStatus.in_progress, | ||
) | ||
report_2 = Report( | ||
robot_id=robot_2.id, | ||
isar_mission_id="isar_mission_id", | ||
echo_mission_id=1, | ||
log="", | ||
status=ReportStatus.completed, | ||
) | ||
|
||
session.add(report_1, report_2) | ||
session.commit() | ||
|
||
entry_1 = Entry( | ||
report_id=report_1.id, | ||
tag_id="tag_id", | ||
status=EntryStatus.completed, | ||
inspection_type=InspectionType.image, | ||
time=datetime.datetime.now(tz=datetime.timezone.utc), | ||
file_location="", | ||
) | ||
|
||
entry_2 = Entry( | ||
report_id=report_2.id, | ||
tag_id="tag_id", | ||
status=EntryStatus.failed, | ||
inspection_type=InspectionType.thermal_image, | ||
time=datetime.datetime.now(tz=datetime.timezone.utc), | ||
file_location="", | ||
) | ||
|
||
entry_3 = Entry( | ||
report_id=report_2.id, | ||
tag_id="tag_id", | ||
status=EntryStatus.completed, | ||
inspection_type=InspectionType.image, | ||
time=datetime.datetime.now(tz=datetime.timezone.utc), | ||
file_location="", | ||
) | ||
|
||
event_1 = Event( | ||
robot_id=robot_1.id, | ||
echo_mission_id=287, | ||
report_id=report_1.id, | ||
estimated_duration=datetime.timedelta(hours=1), | ||
) | ||
|
||
event_2 = Event( | ||
robot_id=robot_2.id, | ||
echo_mission_id=287, | ||
report_id=report_2.id, | ||
estimated_duration=datetime.timedelta(hours=2), | ||
) | ||
|
||
capability_1 = Capability( | ||
robot_id=robot_2.id, | ||
capability=InspectionType.image, | ||
) | ||
|
||
capability_2 = Capability( | ||
robot_id=robot_2.id, | ||
capability=InspectionType.thermal_image, | ||
) | ||
|
||
capability_3 = Capability( | ||
robot_id=robot_1.id, | ||
capability=InspectionType.image, | ||
) | ||
|
||
topic_1 = Topic( | ||
robot_id=robot_1.id, | ||
path="/robot_1/pose", | ||
resource=Resource.pose | ||
) | ||
|
||
topic_2 = Topic( | ||
robot_id=robot_1.id, | ||
path="/robot_1/battery", | ||
resource=Resource.battery | ||
) | ||
|
||
topic_3 = Topic( | ||
robot_id=robot_2.id, | ||
path="/robot_2/pressure", | ||
resource=Resource.pressure | ||
) | ||
|
||
session.add_all( | ||
[ | ||
entry_1, | ||
entry_2, | ||
entry_3, | ||
event_1, | ||
event_2, | ||
capability_1, | ||
capability_2, | ||
capability_3, | ||
topic_1, | ||
topic_2, | ||
topic_3 | ||
] | ||
) | ||
session.commit() |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,118 @@ | ||
import datetime | ||
import enum | ||
import resource | ||
|
||
from sqlalchemy import Column, DateTime, Enum, ForeignKey, Integer, Interval, String | ||
from sqlalchemy.orm import backref, relationship | ||
|
||
from flotilla.database.db import Base | ||
|
||
|
||
class RobotStatus(enum.Enum): | ||
available = "Available" | ||
busy = "Busy" | ||
offline = "Offline" | ||
|
||
|
||
class InspectionType(enum.Enum): | ||
image = "Image" | ||
thermal_image = "ThermalImage" | ||
|
||
|
||
class ReportStatus(enum.Enum): | ||
in_progress = "InProgress" | ||
failed = "Failed" | ||
completed = "Completed" | ||
|
||
|
||
class EntryStatus(enum.Enum): | ||
failed = "Failed" | ||
completed = "Completed" | ||
|
||
class Resource(enum.Enum): | ||
pose = "pose" | ||
battery = "battery" | ||
pressure = "pressure" | ||
|
||
|
||
class Robot(Base): | ||
__tablename__ = "robot" | ||
id = Column(Integer, primary_key=True) | ||
name = Column(String) | ||
model = Column(String) | ||
serial_number = Column(String) | ||
logs = Column(String) | ||
status = Column(Enum(RobotStatus)) | ||
telemetry_topics = relationship("Topic", backref=backref("robot")) | ||
streams = relationship("VideoStream", backref=backref("robot")) | ||
capabilities = relationship("Capability", backref=backref("robot")) | ||
events = relationship("Event", backref=backref("robot")) | ||
|
||
|
||
class Report(Base): | ||
__tablename__ = "report" | ||
id = Column(Integer, primary_key=True) | ||
robot_id = Column(Integer, ForeignKey("robot.id")) | ||
isar_mission_id = Column(String) | ||
echo_mission_id = Column(Integer) | ||
log = Column(String) | ||
status = Column(Enum(ReportStatus)) | ||
entries = relationship("Entry", backref=backref("report")) | ||
|
||
|
||
class Map(Base): | ||
__tablename__ = "map" | ||
id = Column(Integer, primary_key=True) | ||
location = Column(String) | ||
coordinate_reference_system = Column(String) | ||
|
||
|
||
class Topic(Base): | ||
__tablename__ = "topic" | ||
id = Column(Integer, primary_key=True) | ||
robot_id = Column(Integer, ForeignKey("robot.id")) | ||
path = Column(String) | ||
resource = Column(Enum(Resource)) | ||
|
||
|
||
class VideoStream(Base): | ||
__tablename__ = "video_stream" | ||
id = Column(Integer, primary_key=True) | ||
robot_id = Column(Integer, ForeignKey("robot.id")) | ||
rtsp_url = Column(String) | ||
port = Column(Integer) | ||
camera_name = Column(String) | ||
|
||
|
||
class Capability(Base): | ||
__tablename__ = "capability" | ||
id = Column(Integer, primary_key=True) | ||
robot_id = Column(Integer, ForeignKey("robot.id")) | ||
capability = Column(Enum(InspectionType)) | ||
|
||
|
||
class Event(Base): | ||
__tablename__ = "event" | ||
id = Column(Integer, primary_key=True) | ||
robot_id = Column(Integer, ForeignKey("robot.id")) | ||
echo_mission_id = Column(Integer) | ||
report_id = Column(Integer, ForeignKey("report.id")) | ||
start_time = Column( | ||
DateTime(timezone=True), default=datetime.datetime.now(tz=datetime.timezone.utc) | ||
) | ||
estimated_duration = Column(Interval) | ||
# TODO: robot_id and report_id.robot_id can now point at different robots. | ||
# Should there be a constraint forcing an event to point at only one robot? | ||
|
||
|
||
class Entry(Base): | ||
__tablename__ = "entry" | ||
id = Column(Integer, primary_key=True) | ||
report_id = Column(Integer, ForeignKey("report.id")) | ||
tag_id = Column(String) | ||
status = Column(Enum(EntryStatus)) | ||
inspection_type = Column(Enum(InspectionType)) | ||
time = Column( | ||
DateTime(timezone=True), default=datetime.datetime.now(tz=datetime.timezone.utc) | ||
) | ||
file_location = Column(String) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -2,6 +2,7 @@ | |
|
||
app = FastAPI() | ||
|
||
|
||
@app.get("/") | ||
async def root(): | ||
return {"message": "Hello World"} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,12 @@ | ||
import pytest | ||
from sqlalchemy.orm import Session | ||
|
||
from flotilla.database.db import Base, SessionLocal, engine | ||
from flotilla.database.mock_database.mock_database import populate_mock_db | ||
|
||
mock_session: Session = SessionLocal() | ||
populate_mock_db(session=mock_session,engine=engine,base=Base) | ||
|
||
@pytest.fixture() | ||
def session() -> Session: | ||
return mock_session |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,5 @@ | ||
from flotilla.database.models import Robot | ||
|
||
|
||
def test_mock_db(session): | ||
assert len(session.query(Robot.name).all()) == 2 |