Make created_on_str always MST

Author: rubik

history/management/commands/compare_perf.py

@@ -1,7 +1,6 @@
-
 from django.core.management.base import BaseCommand
 from django.conf import settings
-from history.tools import get_fee_amount
+from history.tools import get_fee_amount, utc_to_mst_str
 from history.models import Price, TradeRecommendation, PerformanceComp
 import datetime
 from django.db import transaction
@@ -55,5 +54,5 @@ def handle(self, *args, **options):
                              weighted_avg_nn_rec=weighted_avg_nn_rec,
                              directionally_same=directionally_same,
                              directionally_same_int=1 if directionally_same else 0,
-                             created_on_str=tr_timerange_end.strftime('%Y-%m-%d %H:%M'))
+                             created_on_str=utc_to_mst_str(tr_timerange_end))
         pc.save()

history/management/commands/pull_balance.py

@@ -1,7 +1,7 @@
-
 from django.core.management.base import BaseCommand
 from django.conf import settings
-from history.tools import get_exchange_rate_to_btc, get_exchange_rate_btc_to_usd, get_deposit_balance
+from history.tools import (get_exchange_rate_to_btc, get_exchange_rate_btc_to_usd, get_deposit_balance,
+                           utc_to_mst_str)
 from history.models import Balance, Trade
 import datetime
 from django.db import transaction
@@ -41,15 +41,13 @@ def handle(self, *args, **options):
                     b = Balance(symbol=ticker,coin_balance=val,btc_balance=btc_val,exchange_to_btc_rate=exchange_rate_coin_to_btc,usd_balance=usd_val,exchange_to_usd_rate=exchange_rate_coin_to_btc,deposited_amount_btc=deposited_amount_btc if ticker =='BTC' else 0.00, deposited_amount_usd=deposited_amount_usd if ticker =='BTC' else 0.00)
                     b.save()
 
-        for b in Balance.objects.filter(date_str='0'):
-            # django timezone stuff , FML
-            b.date_str = datetime.datetime.strftime(b.created_on - datetime.timedelta(hours=int(7)),'%Y-%m-%d %H:%M')
+        for b in Balance.objects.filter(date_str=''):
+            b.date_str = utc_to_mst_str(b.created_on)
             b.save()
 
         #normalize trade recommendations too.  merp
         for tr in Trade.objects.filter(created_on_str=''):
-            # django timezone stuff , FML
-            tr.created_on_str = datetime.datetime.strftime(tr.created_on - datetime.timedelta(hours=int(7)),'%Y-%m-%d %H:%M')
+            tr.created_on_str = utc_to_mst_str(tr.created_on)
             tr.save()
 
 

history/management/commands/pull_deposits.py

@@ -1,10 +1,9 @@
-
 from django.core.management.base import BaseCommand
 from django.conf import settings
 import datetime
 from django.db import transaction
 import time
-from history.tools import get_utc_unixtime
+from history.tools import get_utc_unixtime, utc_to_mst_str
 from history.models import Deposit
 
 class Command(BaseCommand):
@@ -40,5 +39,5 @@ def handle(self, *args, **options):
             d.status = status
             d.created_on = created_on
             d.modified_on = created_on
-            d.created_on_str = datetime.datetime.strftime(created_on - datetime.timedelta(hours=int(7)),'%Y-%m-%d %H:%M')
+            d.created_on_str = utc_to_mst_str(created_on)
             d.save()

history/management/commands/pull_prices.py

@@ -1,4 +1,4 @@
-
+from history.tools import utc_to_mst_str
 from django.core.management.base import BaseCommand
 from django.conf import settings
 
@@ -25,6 +25,6 @@ def handle(self, *args, **options):
             p.lowestask = price[ticker]['lowestAsk']
             p.highestbid = price[ticker]['highestBid']
             p.symbol = ticker
-            p.created_on_str = str(p.created_on)
+            p.created_on_str = utc_to_mst_str(p.created_on)
             p.save()
 

history/management/commands/trade.py

@@ -2,7 +2,7 @@
 from history.predict import predict_v2
 from history.models import (Price, PredictionTest, Trade, TradeRecommendation,
                             Balance, ClassifierTest)
-from history.tools import get_utc_unixtime, print_and_log
+from history.tools import get_utc_unixtime, print_and_log, utc_to_mst_str
 import datetime
 import time
 from history.poloniex import poloniex
@@ -187,7 +187,7 @@ def run_predictor(self, nn_index):
                                  confidence=confidence,
                                  recommendation=recommend,
                                  net_amount=-1 if recommend == 'SELL' else (1 if recommend == 'BUY' else 0),
-                                 created_on_str=timezone.now().strftime('%Y-%m-%d %H:%M'))
+                                 created_on_str=utc_to_mst_str(timezone.now()))
         tr.save()
         self.trs[nn_index] = tr
         return recommend

history/tools.py

@@ -1,7 +1,9 @@
 import time
+import pytz
 import datetime
-from django.conf import settings
 import numpy as np
+from django.conf import settings
+from django.utils import timezone
 from django.core.exceptions import ImproperlyConfigured
 
 
@@ -12,14 +14,17 @@ def print_and_log(log_string):
 
 
 def get_utc_unixtime():
-    import time
-    import datetime
-
     d = datetime.datetime.now()
     unixtime = time.mktime(d.timetuple())
     return int(unixtime)
 
 
+def utc_to_mst_str(dt):
+    mst = pytz.timezone('MST')
+    local = timezone.localtime(dt, mst)
+    return datetime.datetime.strftime(local, '%Y-%m-%d %H:%M')
+
+
 def create_sample_row(data, i, size):
     sample = ()
     for k in range(0, size):