humanprotocol
diff --git a/‎packages/examples/cvat/exchange-oracle/Dockerfile‎
Lines changed: 3 additions & 2 deletions b/‎packages/examples/cvat/exchange-oracle/Dockerfile‎
Lines changed: 3 additions & 2 deletions
diff --git a/‎packages/examples/cvat/exchange-oracle/alembic/versions/2cbc85686054_add_assignment_updated_at.py‎
Lines changed: 88 additions & 0 deletions b/‎packages/examples/cvat/exchange-oracle/alembic/versions/2cbc85686054_add_assignment_updated_at.py‎
Lines changed: 88 additions & 0 deletions
diff --git a/‎packages/examples/cvat/exchange-oracle/docker-compose.dev.yml‎
Lines changed: 78 additions & 3 deletions b/‎packages/examples/cvat/exchange-oracle/docker-compose.dev.yml‎
Lines changed: 78 additions & 3 deletions
diff --git a/‎packages/examples/cvat/exchange-oracle/docker-compose.test.yml‎
Lines changed: 17 additions & 2 deletions b/‎packages/examples/cvat/exchange-oracle/docker-compose.test.yml‎
Lines changed: 17 additions & 2 deletions
diff --git a/‎packages/examples/cvat/exchange-oracle/dockerfiles/test.Dockerfile‎
Lines changed: 4 additions & 3 deletions b/‎packages/examples/cvat/exchange-oracle/dockerfiles/test.Dockerfile‎
Lines changed: 4 additions & 3 deletions
@@ -4,12 +4,13 @@ WORKDIR /app
 
 RUN apt-get update -y && \
     apt-get install -y jq ffmpeg libsm6 libxext6 && \
-    pip install poetry
+    pip install --no-cache poetry
 
 COPY pyproject.toml poetry.lock ./
 
 RUN poetry config virtualenvs.create false \
-    && poetry install --no-interaction --no-ansi --no-root
+    && poetry install --no-interaction --no-ansi --no-root \
+    && poetry cache clear pypi --all
 
 COPY . .
 
 
@@ -0,0 +1,88 @@
+"""add-assignment-updated-at
+
+Revision ID: 2cbc85686054
+Revises: 0f3fb7bfcbcf
+Create Date: 2024-09-24 19:18:23.248579
+
+"""
+
+import enum
+
+import sqlalchemy as sa
+from sqlalchemy import Column, DateTime, Enum, String
+from sqlalchemy.orm import Session, declarative_base
+from sqlalchemy.sql import func
+
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision = "2cbc85686054"
+down_revision = "0f3fb7bfcbcf"
+branch_labels = None
+depends_on = None
+
+
+Base = declarative_base()
+
+
+class AssignmentStatuses(str, enum.Enum):
+    created = "created"
+    completed = "completed"
+    expired = "expired"
+    rejected = "rejected"
+    canceled = "canceled"
+
+
+class Assignment(Base):
+    # Represents the model before the transaction is applied
+
+    __tablename__ = "assignments"
+    id = Column(String, primary_key=True, index=True)
+    created_at = Column(DateTime(timezone=True), server_default=func.now())
+    updated_at = Column(DateTime(timezone=True), onupdate=func.now(), server_default=func.now())
+    expires_at = Column(DateTime(timezone=True), nullable=False)
+    completed_at = Column(DateTime(timezone=True), nullable=True, server_default=None)
+    status = Column(
+        String,
+        Enum(AssignmentStatuses),
+        server_default=AssignmentStatuses.created.value,
+        nullable=False,
+    )
+
+
+def define_initial_updated_at():
+    bind = op.get_bind()
+    session = Session(bind=bind)
+
+    session.query(Assignment).filter(
+        Assignment.updated_at == None,
+        Assignment.status.in_(
+            [AssignmentStatuses.expired, AssignmentStatuses.rejected, AssignmentStatuses.canceled]
+        ),
+    ).update({Assignment.updated_at: Assignment.expires_at})
+
+    session.query(Assignment).filter(
+        Assignment.updated_at == None,
+        Assignment.status == AssignmentStatuses.completed,
+    ).update({Assignment.updated_at: Assignment.completed_at})
+
+    session.query(Assignment).filter(
+        Assignment.updated_at == None,
+        # fallback for invalid entries above + handling of status == "created"
+    ).update({Assignment.updated_at: Assignment.created_at})
+
+
+def upgrade() -> None:
+    # This operation can fail because of null values, it's intentional.
+    # Invalid entries in the DB must be fixed in order to proceed
+    op.alter_column("assignments", "created_at", nullable=False)
+
+    op.add_column("assignments", sa.Column("updated_at", sa.DateTime(timezone=True), nullable=True))
+    define_initial_updated_at()
+    op.alter_column("assignments", "updated_at", server_default=sa.text("now()"), nullable=False)
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column("assignments", "updated_at")
+    # ### end Alembic commands ###
@@ -1,7 +1,6 @@
-version: '3.8'
-
 services:
   postgres:
+    container_name: oracle-postgres
     image: postgres:14.4
     restart: on-failure
     environment:
@@ -15,5 +14,81 @@ services:
       - 5431:5432
     command: ["postgres", "-c", "log_statement=all"]
 
+  redis:
+    container_name: oracle-redis
+    image: redis:7.2.3-alpine
+    restart: always
+    command: [
+      "redis-server",
+      "--save", "60", "100",
+      "--appendonly", "yes",
+      # Uncomment if you want to configure a password
+      # "--requirepass", "password",
+      # "--user", "username", "on", ">password", "~*", "allcommands",
+      # "--user", "default", "off", "nopass", "nocommands",
+    ]
+    volumes:
+      - redis:/data
+    ports:
+      - 6380:6379
+    networks:
+      - test-network
+
+  minio:
+    container_name: minio
+    image: minio/minio:RELEASE.2022-05-26T05-48-41Z
+    environment:
+      MINIO_ROOT_USER: dev
+      MINIO_ROOT_PASSWORD: devdevdev
+    volumes:
+      - ./tests/utils/datasets:/tmp/datasets
+    entrypoint: 'sh'
+    ports:
+      - 9010:9010
+      - 9011:9011
+    command:
+      -c "
+      cp -r /tmp/datasets /data &&
+      minio server /data --console-address ':9011' --address ':9010'"
+    healthcheck:
+      test: ["CMD", "curl", "-f", "http://localhost:9010/minio/health/live"]
+      interval: 5s
+      timeout: 5s
+      retries: 3
+    networks:
+      - test-network
+      - cvat-human-bridge
+
+  minio-mc:
+    container_name: minio-mc
+    image: minio/mc
+    depends_on:
+      minio:
+        condition: service_healthy
+    entrypoint: >
+      /bin/sh -c "
+      /usr/bin/mc config host add myminio http://minio:9010 dev devdevdev;
+      /usr/bin/mc anonymous set public myminio/manifests;
+      /usr/bin/mc mb myminio/excor-data;
+      /usr/bin/mc anonymous set public myminio/excor-data;
+      /usr/bin/mc mb myminio/recor-data;
+      /usr/bin/mc anonymous set public myminio/recor-data;
+      /usr/bin/mc mb myminio/datasets;
+      /usr/bin/mc anonymous set public myminio/datasets;
+      /usr/bin/mc mb myminio/launcher;
+      /usr/bin/mc anonymous set public myminio/launcher;
+      "
+    networks:
+      - test-network
+
 volumes:
-  postgres:
+  postgres:
+  redis:
+
+networks:
+  test-network:
+  cvat-human-bridge:
+    name: cvat-human-bridge
+    external: true
+    # Create with:
+    # docker network create -o "com.docker.network.bridge.enable_icc=true"  -o "com.docker.network.driver.mtu=1500" -o "com.docker.network.bridge.enable_ip_masquerade=true" cvat-human-bridge
@@ -1,5 +1,3 @@
-version: '3.8'
-
 services:
   postgres:
     image: postgres:14.4
@@ -14,6 +12,20 @@ services:
     ports:
       - 5432:5432
 
+  redis:
+    container_name: oracle-redis
+    image: redis:7.2.3-alpine
+    restart: always
+    command: [
+      "redis-server",
+      "--save", "60", "100",
+      "--appendonly", "yes",
+    ]
+    ports:
+      - 6379:6379
+    networks:
+      - test-network
+
   minio:
     container_name: minio
     image: minio/minio:RELEASE.2022-05-26T05-48-41Z
@@ -72,9 +84,12 @@ services:
       STORAGE_USE_SSL: False
       STORAGE_PROVIDER: 'aws'
       ENABLE_CUSTOM_CLOUD_HOST: Yes
+      REDIS_HOST: 'redis'
     depends_on:
       postgres:
         condition: service_started
+      redis:
+        condition: service_started
       minio-mc:
         condition: service_completed_successfully
     # Used to reference localhost since minio:9000 is rejected by the SDK URL validator
 
@@ -4,13 +4,14 @@ WORKDIR /app
 
 RUN apt-get update -y && \
     apt-get install -y jq ffmpeg libsm6 libxext6 && \
-    pip install poetry
+    pip install --no-cache poetry
 
 COPY pyproject.toml poetry.lock ./
 
 RUN poetry config virtualenvs.create false \
-    && poetry install --no-interaction --no-ansi --no-root
+    && poetry install --no-interaction --no-ansi --no-root \
+    && poetry cache clear pypi --all
 
 COPY . .
 
-CMD ["pytest", "-W", "ignore::DeprecationWarning"]
+CMD ["pytest", "-W", "ignore::DeprecationWarning", "-v"]