fixing backtesting and perisstence

freqtrade/aws/tables.py

@@ -0,0 +1,93 @@
+import os
+
+import boto3
+from boto.dynamodb2.exceptions import ResourceInUseException
+
+db = boto3.resource('dynamodb')
+
+
+def get_trade_table():
+    """
+        provides access to the trade table and if it doesn't exists
+        creates it for us
+    :return:
+    """
+
+    table_name = os.environ['tradeTable']
+    existing_tables = boto3.client('dynamodb').list_tables()['TableNames']
+    if table_name not in existing_tables:
+        try:
+            db.create_table(
+                TableName=table_name,
+                KeySchema=[
+                    {
+                        'AttributeName': 'id',
+                        'KeyType': 'HASH'
+                    },
+                    {
+                        'AttributeName': 'trade',
+                        'KeyType': 'RANGE'
+                    }
+                ],
+                AttributeDefinitions=[
+                    {
+                        'AttributeName': 'id',
+                        'AttributeType': 'S'
+                    }, {
+                        'AttributeName': 'trade',
+                        'AttributeType': 'S'
+                    }
+                ],
+
+                ProvisionedThroughput={
+                    'ReadCapacityUnits': 1,
+                    'WriteCapacityUnits': 1
+                }
+            )
+        except ResourceInUseException as e:
+            print("table already exist {}".format(e))
+
+    return db.Table(table_name)
+
+
+def get_strategy_table():
+    """
+        provides us access to the strategy table and if it doesn't exists creates it for us
+    :return:
+    """
+    table_name = os.environ['strategyTable']
+    existing_tables = boto3.client('dynamodb').list_tables()['TableNames']
+
+    existing_tables = boto3.client('dynamodb').list_tables()['TableNames']
+    if table_name not in existing_tables:
+        try:
+            db.create_table(
+                TableName=os.environ[table_name],
+                KeySchema=[
+                    {
+                        'AttributeName': 'user',
+                        'KeyType': 'HASH'
+                    },
+                    {
+                        'AttributeName': 'name',
+                        'KeyType': 'RANGE'
+                    }
+                ],
+                AttributeDefinitions=[
+                    {
+                        'AttributeName': 'user',
+                        'AttributeType': 'S'
+                    }, {
+                        'AttributeName': 'name',
+                        'AttributeType': 'S'
+                    }
+                ],
+                ProvisionedThroughput={
+                    'ReadCapacityUnits': 1,
+                    'WriteCapacityUnits': 1
+                }
+            )
+        except ResourceInUseException as e:
+            print("table already exist {}".format(e))
+
+    return db.Table(table_name)