reorganizing modules and scripts

2026-07-02 08:27:53 +08:00 · 2012-02-07 22:16:25 -05:00
parent 4365852b92
commit cca65808d5
101 changed files with 10020 additions and 0 deletions
@@ -0,0 +1,54 @@
+import tornado.auth
+import tornado.httpserver
+import tornado.ioloop
+from tornado.options import define, options
+import tornado.web
+import pymongo
+import bson
+import hashlib
+import base64
+import uuid
+import os
+import logging
+import datetime
+import random
+
+import simulator.qbt_server as qbt_server
+
+MINUTE_COUNT=390
+
+define("user_email", default="qbt@quantopian.com", help="email address for qbt user")
+define("password", default="foobar", help="password for qbt user")
+
+def db_main():
+    tornado.options.parse_command_line()
+    connection, db = qbt_server.connect_db()
+    
+    #create a user for testing
+    salt, encrypted_password = qbt_server.encrypt_password(None, options.password)
+    
+    if not db.users.find_one({'email':options.user_email}):
+        db.users.insert({'email':options.user_email, 'encrypted_password':encrypted_password, 'salt':salt})
+    
+    #create one mythical company
+    if not db.company_info.find_one({'sid':133}):
+        db.company_info.insert({'sid':133, "exchange" : "NEW YORK STOCK EXCHANGE", "symbol" : "JHF", "first date" : "01/04/1993", "last date" : "10/01/2008", "sid" : 133, "industry code" : "130A", "company name" : "JACK INC"})    
+    
+    #create one mythical company
+    if not db.company_info.find_one({'sid':134}):
+        db.company_info.insert({'sid':134, "exchange" : "NEW YORK STOCK EXCHANGE", "symbol" : "RCF", "first date" : "01/04/1993", "last date" : "10/01/2008", "sid" : 134, "industry code" : "130A", "company name" : "ROCCO INC"})    
+        
+    #create minute equity data collection and populate with a day of random data
+    prices = {133:25.0,134:45.0} #sid, initial price.
+    if not db.equity.trades.minute.find().count() == MINUTE_COUNT * len(prices):
+        db.equity.trades.minute.drop()
+        trade_start = datetime.datetime.now()
+        minute = datetime.timedelta(minutes=1)
+
+        for i in range(MINUTE_COUNT):
+            for sid,price in prices.iteritems():
+                price = price + random.uniform(-0.05,0.05)
+                db.equity.trades.minute.insert({'sid':sid, 'dt':trade_start + (minute * i),'price':price, 'volume':random.randrange(100,10000,100)})
+
+if __name__ == "__main__":
+    db_main()