dao-quantum-identity/models.py at main · Sampo-AI-OS/dao-quantum-identity · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
import uuid
from datetime import datetime, timezone

from sqlalchemy import Boolean, Column, DateTime, ForeignKey, Integer, String, Text
from sqlalchemy.orm import relationship

from database import Base


def _utc_now():
    return datetime.now(timezone.utc)


class User(Base):
    __tablename__ = "users"

    id = Column(Integer, primary_key=True, index=True)
    username = Column(String(100), unique=True, nullable=False)
    email = Column(String(200), nullable=False)
    hashed_password = Column(String(300), nullable=False)
    role = Column(String(50), default="user")
    is_active = Column(Boolean, default=True)
    created_at = Column(DateTime(timezone=True), default=_utc_now)


class DecentralizedIdentity(Base):
    __tablename__ = "decentralized_identities"

    id = Column(Integer, primary_key=True, index=True)
    did = Column(String(300), unique=True, nullable=False)
    controller = Column(String(200), nullable=False)
    subject_name = Column(String(200), nullable=False)
    pq_algorithm = Column(String(100), nullable=False)
    public_key_fingerprint = Column(String(128), nullable=False)
    public_key_pem = Column(Text, nullable=False)
    did_document_json = Column(Text, nullable=False)
    status = Column(String(50), default="active")
    blockchain_anchor_hash = Column(String(64), nullable=True)
    blockchain_ledger = Column(String(100), nullable=True)
    created_at = Column(DateTime(timezone=True), default=_utc_now)
    updated_at = Column(DateTime(timezone=True), default=_utc_now)
    credentials = relationship("VerifiableCredential", back_populates="subject_did_ref", foreign_keys="VerifiableCredential.subject_did")
    presentations = relationship("ZKPresentation", back_populates="holder_did_ref", cascade="all, delete-orphan")


class VerifiableCredential(Base):
    __tablename__ = "verifiable_credentials"

    id = Column(Integer, primary_key=True, index=True)
    credential_id = Column(String(300), unique=True, nullable=False)
    issuer_did = Column(String(300), nullable=False)
    subject_did = Column(String(300), ForeignKey("decentralized_identities.did"), nullable=False)
    credential_type = Column(String(100), nullable=False)
    schema_id = Column(String(200), nullable=True)
    claims_json = Column(Text, nullable=False)
    claims_hash = Column(String(64), nullable=False)
    signature = Column(Text, nullable=False)
    pq_algorithm = Column(String(100), nullable=False)
    issuance_date = Column(DateTime(timezone=True), default=_utc_now)
    expiry_date = Column(DateTime(timezone=True), nullable=True)
    status = Column(String(50), default="active")
    revoked_at = Column(DateTime(timezone=True), nullable=True)
    revocation_reason = Column(String(300), nullable=True)
    blockchain_anchor_hash = Column(String(64), nullable=True)
    subject_did_ref = relationship("DecentralizedIdentity", back_populates="credentials", foreign_keys=[subject_did])
    zk_commitments = relationship("ZKCommitment", back_populates="credential", cascade="all, delete-orphan")


class ZKCommitment(Base):
    __tablename__ = "zk_commitments"

    id = Column(Integer, primary_key=True, index=True)
    credential_id = Column(String(300), ForeignKey("verifiable_credentials.credential_id"), nullable=False)
    attribute_name = Column(String(200), nullable=False)
    commitment_hash = Column(String(64), nullable=False)
    salt = Column(String(64), nullable=False)
    is_selectively_disclosable = Column(Boolean, default=True)
    created_at = Column(DateTime(timezone=True), default=_utc_now)
    credential = relationship("VerifiableCredential", back_populates="zk_commitments")


class ZKPresentation(Base):
    __tablename__ = "zk_presentations"

    id = Column(Integer, primary_key=True, index=True)
    presentation_id = Column(String(300), unique=True, nullable=False)
    holder_did = Column(String(300), ForeignKey("decentralized_identities.did"), nullable=False)
    verifier_id = Column(String(300), nullable=False)
    credential_ids_json = Column(Text, nullable=False)
    disclosed_attributes_json = Column(Text, nullable=False)
    undisclosed_commitments_json = Column(Text, nullable=False)
    proof_json = Column(Text, nullable=False)
    is_verified = Column(Boolean, default=False)
    verification_result_json = Column(Text, nullable=True)
    expires_at = Column(DateTime(timezone=True), nullable=True)
    created_at = Column(DateTime(timezone=True), default=_utc_now)
    holder_did_ref = relationship("DecentralizedIdentity", back_populates="presentations")


class BlockchainAnchor(Base):
    __tablename__ = "blockchain_anchors"

    id = Column(Integer, primary_key=True, index=True)
    entity_type = Column(String(50), nullable=False)
    entity_id = Column(String(300), nullable=False)
    ledger = Column(String(100), nullable=False)
    transaction_hash = Column(String(64), nullable=False)
    block_hash = Column(String(64), nullable=False)
    block_number = Column(Integer, nullable=True)
    prev_anchor_hash = Column(String(64), nullable=True)
    data_hash = Column(String(64), nullable=False)
    anchored_at = Column(DateTime(timezone=True), default=_utc_now)


class RevocationRegistry(Base):
    __tablename__ = "revocation_registry"

    id = Column(Integer, primary_key=True, index=True)
    credential_id = Column(String(300), unique=True, nullable=False)
    issuer_did = Column(String(300), nullable=False)
    subject_did = Column(String(300), nullable=False)
    revocation_type = Column(String(50), default="immediate")
    reason = Column(String(300), nullable=True)
    revocation_hash = Column(String(64), nullable=False)
    revoked_at = Column(DateTime(timezone=True), default=_utc_now)