Remove SQS related features

finance/__main__.py

@@ -33,7 +33,6 @@
     insert_stock_record,
     parse_date,
     parse_stock_records,
-    request_import_stock_values as request_import_stock_values_,
     serialize_datetime,
 )
 
@@ -315,15 +314,5 @@ def import_stock_records(filename):
             insert_stock_record(parsed, account_stock, account_bank)
 
 
-@cli.command()
-@click.argument("code")
-def request_import_stock_values(code):
-    """Enqueue a request to import stock values."""
-    start_time = date_to_datetime(parse_date(-3))
-    end_time = date_to_datetime(parse_date(0))
-
-    request_import_stock_values_(code, start_time, end_time)
-
-
 if __name__ == "__main__":
     cli()

finance/utils.py

@@ -245,36 +245,6 @@ def parse_stock_records(stream):
         }
 
 
-def poll_import_stock_values_requests(sqs_region, queue_url):
-    client = boto3.client("sqs", region_name=sqs_region)
-    resp = client.receive_message(**{"QueueUrl": queue_url, "VisibilityTimeout": 180})
-    messages = resp["Messages"] if "Messages" in resp else []
-
-    for message in messages:
-        yield json.loads(message["Body"])
-        client.delete_message(
-            **{"QueueUrl": queue_url, "ReceiptHandle": message["ReceiptHandle"]}
-        )
-
-
-def request_import_stock_values(
-    code,
-    start_time,
-    end_time,
-    sqs_region=os.environ.get("SQS_REGION", ""),
-    queue_url=os.environ.get("REQUEST_IMPORT_STOCK_VALUES_QUEUE_URL", ""),
-):
-    message = make_request_import_stock_values_message(code, start_time, end_time)
-
-    client = boto3.client("sqs", region_name=sqs_region)
-    resp = client.send_message(
-        **{"QueueUrl": queue_url, "MessageBody": json.dumps(message),}
-    )
-
-    if resp["ResponseMetadata"]["HTTPStatusCode"] != 200:
-        log.error("Something went wrong: {0}", resp)
-
-
 def insert_stock_record(data: dict, stock_account: object, bank_account: object):
     if data["category2"] in ["매도", "매수"]:
         return insert_stock_trading_record(data, stock_account)

infra/lambda.py

@@ -12,8 +12,6 @@
 from finance.utils import (
     date_to_datetime,
     parse_date,
-    poll_import_stock_values_requests,
-    request_import_stock_values,
 )
 
 
@@ -23,32 +21,6 @@
 # TODO: Write logs to CloudWatch
 
 
-def request_import_stock_values_handler(event, context):
-    codes = ["AMD", "AMZN", "BRK-A", "BRK-B", "ESRT", "NVDA", "SBUX", "SPY"]
-    start_time = date_to_datetime(parse_date(-3))
-    end_time = date_to_datetime(parse_date(0))
-
-    for code in codes:
-        request_import_stock_values(code, start_time, end_time)
-
-    log.info("Requested to import stock values: {0}", ", ".join(codes))
-
-
-def fetch_asset_values_handler(event, context):
-    config = {"SQLALCHEMY_DATABASE_URI": os.environ["SBF_DB_URL"]}
-    sqs_region = os.environ["SQS_REGION"]
-    queue_url = os.environ["REQUEST_IMPORT_STOCK_VALUES_QUEUE_URL"]
-
-    app = create_app(__name__, config=config)
-    with app.app_context():
-        requests = poll_import_stock_values_requests(sqs_region, queue_url)
-        for request in requests:
-            code = request["code"]
-            start_time = datetime.fromtimestamp(request["start_time"])
-            end_time = datetime.fromtimestamp(request["end_time"])
-            fetch_asset_values(code, start_time, end_time)
-
-
 def fetch_asset_values(code, start_time, end_time):
     try:
         asset = Asset.get_by_symbol(code)
@@ -100,4 +72,3 @@ def insert_asset_value(
 
 if __name__ == "__main__":
     request_import_stock_values_handler({}, None)
-    # fetch_asset_values_handler({}, None)

provision.sh.dist

@@ -9,5 +9,4 @@ export AWS_DEFAULT_REGION="us-west-2"
 export GSPREAD_KEY_FILE_PATH="keyfile.json"
 export GSPREAD_DOC_KEY=""
 
-export SQS_REGION="us-west-2"
 export REQUEST_IMPORT_STOCK_VALUES_QUEUE_URL=""