Fix formatting with black
This commit is contained in:
@ -1,4 +1,13 @@
|
||||
from sqlalchemy import Column, Integer, String, Date, ForeignKey, JSON, DateTime, Boolean
|
||||
from sqlalchemy import (
|
||||
Column,
|
||||
Integer,
|
||||
String,
|
||||
Date,
|
||||
ForeignKey,
|
||||
JSON,
|
||||
DateTime,
|
||||
Boolean,
|
||||
)
|
||||
from sqlalchemy.orm import relationship
|
||||
from .database import Base
|
||||
from datetime import datetime
|
||||
@ -14,7 +23,7 @@ class Shipment(Base):
|
||||
comments = Column(String(200), nullable=True)
|
||||
contact_person_id = Column(Integer, ForeignKey("contact_persons.id"))
|
||||
return_address_id = Column(Integer, ForeignKey("addresses.id"))
|
||||
proposal_id = Column(Integer, ForeignKey('proposals.id'), nullable=True)
|
||||
proposal_id = Column(Integer, ForeignKey("proposals.id"), nullable=True)
|
||||
|
||||
contact_person = relationship("ContactPerson", back_populates="shipments")
|
||||
return_address = relationship("Address", back_populates="shipments")
|
||||
@ -44,17 +53,19 @@ class Address(Base):
|
||||
|
||||
shipments = relationship("Shipment", back_populates="return_address")
|
||||
|
||||
|
||||
class DewarType(Base):
|
||||
__tablename__ = "dewar_types"
|
||||
id = Column(Integer, primary_key=True, index=True, autoincrement=True)
|
||||
dewar_type = Column(String(255), unique=True, index=True)
|
||||
serial_numbers = relationship("DewarSerialNumber", back_populates="dewar_type")
|
||||
|
||||
|
||||
class DewarSerialNumber(Base):
|
||||
__tablename__ = "dewar_serial_numbers"
|
||||
id = Column(Integer, primary_key=True, index=True, autoincrement=True)
|
||||
serial_number = Column(String(255), index=True)
|
||||
dewar_type_id = Column(Integer, ForeignKey('dewar_types.id'))
|
||||
dewar_type_id = Column(Integer, ForeignKey("dewar_types.id"))
|
||||
dewar_type = relationship("DewarType", back_populates="serial_numbers")
|
||||
|
||||
|
||||
@ -64,7 +75,9 @@ class Dewar(Base):
|
||||
id = Column(Integer, primary_key=True, index=True, autoincrement=True)
|
||||
dewar_name = Column(String(255))
|
||||
dewar_type_id = Column(Integer, ForeignKey("dewar_types.id"), nullable=True)
|
||||
dewar_serial_number_id = Column(Integer, ForeignKey("dewar_serial_numbers.id"), nullable=True)
|
||||
dewar_serial_number_id = Column(
|
||||
Integer, ForeignKey("dewar_serial_numbers.id"), nullable=True
|
||||
)
|
||||
tracking_number = Column(String(255))
|
||||
status = Column(String(255))
|
||||
ready_date = Column(Date, nullable=True)
|
||||
@ -97,6 +110,7 @@ class Dewar(Base):
|
||||
return 0
|
||||
return sum(len(puck.samples) for puck in self.pucks)
|
||||
|
||||
|
||||
class Proposal(Base):
|
||||
__tablename__ = "proposals"
|
||||
|
||||
@ -106,7 +120,7 @@ class Proposal(Base):
|
||||
|
||||
|
||||
class Puck(Base):
|
||||
__tablename__ = 'pucks'
|
||||
__tablename__ = "pucks"
|
||||
|
||||
id = Column(Integer, primary_key=True, index=True, autoincrement=True)
|
||||
puck_name = Column(String(255), index=True)
|
||||
@ -114,14 +128,14 @@ class Puck(Base):
|
||||
puck_location_in_dewar = Column(Integer)
|
||||
|
||||
# Foreign keys and relationships
|
||||
dewar_id = Column(Integer, ForeignKey('dewars.id'))
|
||||
dewar_id = Column(Integer, ForeignKey("dewars.id"))
|
||||
dewar = relationship("Dewar", back_populates="pucks")
|
||||
samples = relationship("Sample", back_populates="puck")
|
||||
events = relationship("PuckEvent", back_populates="puck")
|
||||
|
||||
|
||||
class Sample(Base):
|
||||
__tablename__ = 'samples'
|
||||
__tablename__ = "samples"
|
||||
|
||||
id = Column(Integer, primary_key=True, index=True, autoincrement=True)
|
||||
sample_name = Column(String(255), index=True)
|
||||
@ -129,7 +143,7 @@ class Sample(Base):
|
||||
data_collection_parameters = Column(JSON, nullable=True)
|
||||
|
||||
# Foreign keys and relationships
|
||||
puck_id = Column(Integer, ForeignKey('pucks.id'))
|
||||
puck_id = Column(Integer, ForeignKey("pucks.id"))
|
||||
puck = relationship("Puck", back_populates="samples")
|
||||
events = relationship("SampleEvent", back_populates="sample")
|
||||
|
||||
@ -143,38 +157,41 @@ class Slot(Base):
|
||||
qr_base = Column(String(255), nullable=True)
|
||||
occupied = Column(Boolean, default=False)
|
||||
needs_refill = Column(Boolean, default=False)
|
||||
dewar_unique_id = Column(String(255), ForeignKey('dewars.unique_id'), nullable=True)
|
||||
dewar_unique_id = Column(String(255), ForeignKey("dewars.unique_id"), nullable=True)
|
||||
dewar = relationship("Dewar", back_populates="slot")
|
||||
events = relationship("LogisticsEvent", back_populates="slot")
|
||||
|
||||
|
||||
class LogisticsEvent(Base):
|
||||
__tablename__ = "logistics_events"
|
||||
|
||||
id = Column(Integer, primary_key=True, index=True)
|
||||
dewar_id = Column(Integer, ForeignKey('dewars.id'))
|
||||
slot_id = Column(Integer, ForeignKey('slots.id'))
|
||||
dewar_id = Column(Integer, ForeignKey("dewars.id"))
|
||||
slot_id = Column(Integer, ForeignKey("slots.id"))
|
||||
event_type = Column(String(255), index=True)
|
||||
timestamp = Column(DateTime, default=datetime.utcnow)
|
||||
dewar = relationship("Dewar", back_populates="events")
|
||||
slot = relationship("Slot", back_populates="events")
|
||||
|
||||
|
||||
class SampleEvent(Base):
|
||||
__tablename__ = "sample_events"
|
||||
|
||||
id = Column(Integer, primary_key=True, index=True)
|
||||
sample_id = Column(Integer, ForeignKey('samples.id'))
|
||||
sample_id = Column(Integer, ForeignKey("samples.id"))
|
||||
event_type = Column(String(255), index=True)
|
||||
timestamp = Column(DateTime, default=datetime.utcnow)
|
||||
|
||||
sample = relationship("Sample", back_populates="events")
|
||||
|
||||
|
||||
class PuckEvent(Base):
|
||||
__tablename__ = "puck_events"
|
||||
|
||||
id = Column(Integer, primary_key=True, index=True)
|
||||
puck_id = Column(Integer, ForeignKey('pucks.id'))
|
||||
puck_id = Column(Integer, ForeignKey("pucks.id"))
|
||||
tell_position = Column(String(255), nullable=True)
|
||||
event_type = Column(String(255), index=True)
|
||||
timestamp = Column(DateTime, default=datetime.utcnow)
|
||||
|
||||
puck = relationship("Puck", back_populates="events")
|
||||
puck = relationship("Puck", back_populates="events")
|
||||
|
Reference in New Issue
Block a user