main_dict.py

Go to the documentation of this file.

#!/usr/bin/python

## @package serverportfolio.main_dict
# @brief Executable to test the module DictionaryStocks.
#
# Because it is defined as a singleton, the main cannot be included in the original module.
#
# Last Changed $Id: main_dict.py 9 2015-04-02 23:27:25Z michael $



#import sys, re, types, logging
import sys, logging, traceback
#import traceback
# bug parser try to sleep
#import time
# try for bug related to split
#import string
#import os
#import os.path
#import time, datetime

# import first the singleton
from serverportfolio.DictionaryStocks import DictionaryStocks

#from serverportfolio.GlobalDicts import *
#from serverportfolio.ModClientTCP import TCPClient
from serverportfolio import Utils
from serverportfolio.PortfolioException import PortfolioError, ParserError


    
# ############# Main only for standalone test ########

# problem because DictStock variable in __name__, cannot access from import module !
if __name__ == "__main__":
    
    import sys
    print "DictionaryStocks main"   
    
    # To use logger need to set in all main executables
    logging.basicConfig(level=logging.DEBUG)
    # write output to a file
    # logging.basicConfig(filename='myapp.log', level=logging.INFO) # filemode='w' to not append to file
    # all modules declare child loggers of SP
    logger = logging.getLogger('SP')
    # logger.warning("warning1")
    # specific to the main part
    main_logger = logging.getLogger("SP.main") 
        
    # try to initialise the dictionary with the file given as first argument, searched in PATH_DATA
    try:
        dict_filename = str(sys.argv[1])
        main.logger.debug("dictionary file as argument: %s" % dict_filename)
        dict_stocks1 = DictionaryStocks( dict_filename )
        
    # if argument not given, use the default dictstocks.txt
    except:
        dict_stocks1 = DictionaryStocks()
    
    print "DictionaryStocks initialized"
    #sys.exit()
    
    print "\n== dict_stocks 2, test Singleton"
    # OK, init is not called this time. Singleton working
    dict_stocks2 = DictionaryStocks( "dictstocks.txt" )
    # test id, should be the same
    print dict_stocks1
    print id(dict_stocks1)
    print dict_stocks2
    print id(dict_stocks2)
    #sys.exit()
    
    # load a new stock
    print "\n== get 1 Stock, CAC40"
    DictionaryStocks().get_stocks('CAC40')
    #sys.exit(1)
    
    # test function, print data
    print "First Print "
    dict_stocks1.print_dict_stocks( opt_print = True )
    #sys.exit(1)
    
    print "Second Print "
    lines = dict_stocks1.print_dict_stocks( "CAC40", opt_print = False )
    print "line CAC40"
    print lines
    
    # ask for a non existing Stock, got the exception
    # no more exception but a InvalidStock
    #try:
    dict_stocks1.get_stocks("toto")
    #except PortfolioError as ex:
    #    print "Got PortfolioError ", ex
    #sys.exit(1)
    
    # print a non-existing Stock
    print "\n== Print a existing and non-loaded stock"
    lines = dict_stocks1.print_dict_stocks( ["CAC40","SP500"], opt_print = False )
    print "line CAC40 SP500"
    print lines
    
    print "\n== Print 2 existing and 1 non-existing stock"
    lines = dict_stocks1.print_dict_stocks( ["CAC40","SP500", "toto"], opt_print = False )
    print "line CAC40 SP500 toto"
    print lines
    #sys.exit()
    
    
    print "\n=== Update Inst Value"
    try :
        #dict_stocks1.update_inst_value()
        # more general
        dict_stocks1.update_data( action='InstValue' )

    except PortfolioError as ex:
        print "Main catch a PortfolioError "
        print "message ", ex # print only exc_value
        # test is instance
        #if ex.url is not None:
        #    print "url from ex ", ex.url
        # from documentation traceback, ok
        exc_type, exc_value, exc_traceback = sys.exc_info()
        # traceback lost again...
        print "*** print_tb:"
        traceback.print_tb(exc_traceback, limit=2, file=sys.stdout)
        print "full message: \n", ex.get_format_string()
        
        print "exit properly "
        sys.exit(1)

    # global exception, should not happen
    except Exception as ex:
        #self.logger.error("got exception in Update() %s " % ex )
        print "got general Exception in Update(), Should not happen %s " % ex
        exc_type, exc_value, exc_traceback = sys.exc_info()
        # traceback lost again...
        print "*** print_tb:"
        traceback.print_tb(exc_traceback, limit=4, file=sys.stdout)
        #print traceback
        sys.exit(1)
    
# post-process
    print '\n== After update inst value, summarize'
    dict_stocks1.print_dict_stocks( opt_print = True )
    #sys.exit(1)
    
    print "\n===== all stocks ===== "
    print dict_stocks1
    dict_stocks1.print_dict_stocks( opt_print = True )
    #sys.exit(1)
    # make different test
    # useful, independent of the structure, dict_stocks.get_liststocks()
    
    print "\n=== check_open "
    #for stock in dict_stocks1.get_stocks( ["CAC40","SP500", "toto"] ):
    for stock in ["CAC40","SP500", "toto"]:
        #print "stock : ", stock
        print "   CheckOpen ",dict_stocks1.check_open(stock)
        #dict_stocks1.check_open( ["CAC40","SP500", "toto"] )
        # time to open, gives strange results
        #time_to_open = dict_stocks1.time_to_open(stock)
        #print "   time_to_open ",time_to_open
        
    ## test to add, TestML_done
    dict_stocks1.print_dict_stocks( opt_print = True )
    #sys.exit(1)
    
    # has been moved to Stock, logic!
    print "\n=== time_to_open "
    for stock in dict_stocks1.get_stocks( ["CAC40","SP500", "toto"] ):
        time_to_open = stock.time_to_open()
        print "   time_to_open ",time_to_open
    dict_stocks1.print_dict_stocks( opt_print = True )
    #sys.exit(1)
    
    ## Test save to file
    print "\n SaveInstValue "
    #for stock in dict_stocks1.GetListStocks():
    for stock in dict_stocks1.get_stocks( ["CAC40","SP500", "toto"] ):
       print "stock : ", stock
       
       # from AutoParser
       # here get last values from the web
       #dict_data=self.dict_stocks.GetInstValue(namestock)
       #self.dict_stocks.SetLastModification ( namestock, dict_data['time'] )
       #self.dict_stocks.AddVolatileData( namestock, dict_data )

       #dict_data = dict_stocks1.GetInstValue( stock )
       #dict_data = dict_stock1.GetStock( stock )
       #dict_stocks1.SetLastModification( stock, dict_data['Static']['time'] )
       #dict_stocks1.AddVolatileData( stock, dict_data )
       
       #dict_stocks1.SaveInstValue( stock, dict_data )
       #dict_stocks1.SaveInstValue2( stock, dict_stocks1[stock]['InstValue'] )
       stock.save_inst_value()