Merge b693b4bfd14039d34d9c5476eecdde88038be771 into ce8179f07e4fb8740b43570aa2c5826447c2af26

Dockerfile

@@ -1,6 +1,6 @@
 FROM python:3.9-slim-buster
 
-ENV POETRY_VERSION=1.1.12
+ENV POETRY_VERSION=1.8.4
 
 RUN useradd --create-home flashbot \
   && apt-get update \
@@ -17,8 +17,7 @@ WORKDIR /app/
 
 USER flashbot
 
-RUN poetry config virtualenvs.create false \
-    && poetry install
+RUN poetry install
 
 COPY --chown=flashbot . /app
 

mev

@@ -4,15 +4,15 @@ set -e
 
 DB_NAME=mev_inspect
 
-function get_kube_secret(){
+function get_kube_secret() {
   kubectl get secrets $1 -o jsonpath="{.data.$2}" | base64 --decode
 }
 
-function get_kube_db_secret(){
+function get_kube_db_secret() {
   kubectl get secrets mev-inspect-db-credentials -o jsonpath="{.data.$1}" | base64 --decode
 }
 
-function db(){
+function db() {
   host=$(get_kube_secret "mev-inspect-db-credentials" "host")
   username=$(get_kube_secret "mev-inspect-db-credentials" "username")
   password=$(get_kube_secret "mev-inspect-db-credentials" "password")
@@ -23,7 +23,7 @@ function db(){
     -- $DB_NAME --host=$host --user=$username
 }
 
-function redis(){
+function redis() {
   echo "To continue, enter 'shift + r'"
   redis_password=$(get_kube_secret "redis" "redis-password")
   kubectl run -i --rm --tty \
@@ -34,51 +34,51 @@ function redis(){
 }
 
 case "$1" in
-  db)
+db)
   echo "Connecting to $DB_NAME"
   db
   ;;
-  redis)
+redis)
   echo "Connecting to redis"
   redis
   ;;
-  listener)
+listener)
   kubectl exec -ti deploy/mev-inspect -- ./listener $2
   ;;
-  block-list)
+block-list)
   echo "Backfilling blocks from stdin"
   kubectl exec -i deploy/mev-inspect -- poetry run enqueue-block-list
   ;;
-  backfill)
+backfill)
   after_block_number=$2
   before_block_number=$3
 
   echo "Backfilling from $after_block_number to $before_block_number"
   kubectl exec -ti deploy/mev-inspect -- poetry run enqueue-many-blocks $after_block_number $before_block_number
   ;;
-  inspect)
+inspect)
   block_number=$2
   echo "Inspecting block $block_number"
   kubectl exec -ti deploy/mev-inspect -- poetry run inspect-block $block_number
   ;;
-  inspect-many)
+inspect-many)
   after_block_number=$2
   before_block_number=$3
   echo "Inspecting from block $after_block_number to $before_block_number"
   kubectl exec -ti deploy/mev-inspect -- \
     poetry run inspect-many-blocks $after_block_number $before_block_number
   ;;
-  test)
+test)
   shift
   echo "Running tests"
   kubectl exec -ti deploy/mev-inspect -- poetry run pytest tests $@
   ;;
-  fetch)
+fetch)
   block_number=$2
   echo "Fetching block $block_number"
   kubectl exec -ti deploy/mev-inspect -- poetry run fetch-block $block_number
   ;;
-  prices)
+prices)
   shift
   case "$1" in
   fetch-all)
@@ -96,34 +96,40 @@ case "$1" in
   *)
     echo "prices usage: "$1" {fetch-all}"
     exit 1
+    ;;
   esac
   ;;
-  backfill-export)
+backfill-export)
   after_block=$2
   before_block=$3
 
   echo "Sending $after_block to $before_block export to queue"
   kubectl exec -ti deploy/mev-inspect -- poetry run enqueue-many-s3-exports $after_block $before_block
   ;;
-  enqueue-s3-export)
+enqueue-s3-export)
   block_number=$2
 
   echo "Sending $block_number export to queue"
   kubectl exec -ti deploy/mev-inspect -- poetry run enqueue-s3-export $block_number
   ;;
-  s3-export)
+s3-export)
   block_number=$2
 
   echo "Exporting $block_number"
   kubectl exec -ti deploy/mev-inspect -- poetry run s3-export $block_number
   ;;
-  exec)
+migrate)
+  echo "Applying database migrations"
+  kubectl exec -ti deploy/mev-inspect -- poetry run alembic upgrade head
+  ;;
+exec)
   shift
   kubectl exec -ti deploy/mev-inspect -- $@
   ;;
-  *)
+*)
   echo "Usage: "$1" {db|backfill|inspect|test}"
   exit 1
+  ;;
 esac
 
 exit 0