79 lines
3.4 KiB
Python
79 lines
3.4 KiB
Python
from sqlalchemy import Column, Integer, String, Text, DateTime, Boolean, Float, ForeignKey
|
|
from sqlalchemy.sql import func
|
|
from app.core.config import Base
|
|
|
|
|
|
class ProviderAccount(Base):
|
|
__tablename__ = 'provider_accounts'
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
provider = Column(String(32), nullable=False, index=True) # anthropic/openai/minimax/kimi/qwen
|
|
label = Column(String(128), nullable=False)
|
|
credential = Column(Text, nullable=False) # TODO: encrypt at rest
|
|
is_enabled = Column(Boolean, default=True)
|
|
created_by = Column(Integer, nullable=True)
|
|
created_at = Column(DateTime(timezone=True), server_default=func.now())
|
|
updated_at = Column(DateTime(timezone=True), onupdate=func.now())
|
|
|
|
|
|
class ProviderUsageSnapshot(Base):
|
|
__tablename__ = 'provider_usage_snapshots'
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
account_id = Column(Integer, ForeignKey('provider_accounts.id'), nullable=False, index=True)
|
|
window_label = Column(String(32), nullable=True) # e.g. 1h / 7d
|
|
used = Column(Float, nullable=True)
|
|
limit = Column(Float, nullable=True)
|
|
usage_pct = Column(Float, nullable=True)
|
|
reset_at = Column(DateTime(timezone=True), nullable=True)
|
|
status = Column(String(32), nullable=False, default='unknown') # ok/error/pending/unsupported
|
|
error = Column(Text, nullable=True)
|
|
raw_payload = Column(Text, nullable=True)
|
|
fetched_at = Column(DateTime(timezone=True), server_default=func.now(), index=True)
|
|
|
|
|
|
class MonitoredServer(Base):
|
|
__tablename__ = 'monitored_servers'
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
identifier = Column(String(128), nullable=False, unique=True)
|
|
display_name = Column(String(128), nullable=True)
|
|
is_enabled = Column(Boolean, default=True)
|
|
created_by = Column(Integer, nullable=True)
|
|
created_at = Column(DateTime(timezone=True), server_default=func.now())
|
|
|
|
|
|
class ServerState(Base):
|
|
__tablename__ = 'server_states'
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
server_id = Column(Integer, ForeignKey('monitored_servers.id'), nullable=False, unique=True)
|
|
openclaw_version = Column(String(64), nullable=True)
|
|
agents_json = Column(Text, nullable=True) # json list
|
|
cpu_pct = Column(Float, nullable=True)
|
|
mem_pct = Column(Float, nullable=True)
|
|
disk_pct = Column(Float, nullable=True)
|
|
swap_pct = Column(Float, nullable=True)
|
|
last_seen_at = Column(DateTime(timezone=True), nullable=True)
|
|
updated_at = Column(DateTime(timezone=True), onupdate=func.now())
|
|
|
|
|
|
class ServerChallenge(Base):
|
|
__tablename__ = 'server_challenges'
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
server_id = Column(Integer, ForeignKey('monitored_servers.id'), nullable=False, index=True)
|
|
challenge_uuid = Column(String(64), nullable=False, unique=True, index=True)
|
|
expires_at = Column(DateTime(timezone=True), nullable=False)
|
|
used_at = Column(DateTime(timezone=True), nullable=True)
|
|
created_at = Column(DateTime(timezone=True), server_default=func.now())
|
|
|
|
|
|
class ServerHandshakeNonce(Base):
|
|
__tablename__ = 'server_handshake_nonces'
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
server_id = Column(Integer, ForeignKey('monitored_servers.id'), nullable=False, index=True)
|
|
nonce = Column(String(128), nullable=False, index=True)
|
|
created_at = Column(DateTime(timezone=True), server_default=func.now())
|