Ported 83_Stock_Market to Python

I've used an Object-Oriented Approach. The code's a little messy, We'll clean it up later.

83_Stock_Market/python/Stock_Market.py

@@ -0,0 +1,208 @@
+import random
+
+#Stock_Market
+class Stock_Market():
+
+  def __init__(self):
+
+    #Hard Coded Names
+    short_names = ['IBM', 'RCA', 'LBJ', 'ABC', 'CBS']
+    full_names = ['INT. BALLISTIC MISSLES', 'RED CROSS OF AMERICA', 
+                  'LICHTENSTEIN, BUMRAP & JOKE', 'AMERICAN BANKRUPT CO.',
+                  'CENSURED BOOKS STORE']
+
+    #Initializing Dictionary to hold all the information systematically
+    self.data = {}
+    for sn, fn in zip(short_names, full_names):
+      #A dictionary for each stock
+      temp = {'Name' : fn, 'Price' : None, 'Holdings' : 0}
+      #Nested outer dictionary for all stocks
+      self.data[sn] = temp
+
+    #Initializing Randomly generated initial prices
+    for stock in self.data.values():
+      stock['Price'] = round(random.uniform(80,120),2) #Price b/w 60 and 120
+
+    #Initialize Assets
+    self.cash_assets = 10000
+    self.stock_assets = 0
+    
+  def total_assets(self):
+    
+    return self.cash_assets + self.stock_assets
+
+  def _generate_day_change(self):
+
+    self.changes = []
+    for _ in range(len(self.data)):
+      self.changes.append(round(random.uniform(-5,5),2)) #Random % Change b/w -5 and 5
+
+  def update_prices(self):
+
+    self._generate_day_change()
+    for stock, change in zip(self.data.values(), self.changes):
+      stock['Price'] = round(stock['Price'] + (change/100)*stock['Price'], 2)
+
+  def print_exchange_average(self):
+    
+    sum = 0
+    for stock in self.data.values():
+      sum += stock['Price']
+
+    print('\nNEW YORK STOCK EXCHANGE AVERAGE: ${:.2f}'.format(sum/5))
+
+  def get_average_change(self):
+
+    sum = 0
+    for change in self.changes:
+      sum += change
+    
+    return round(sum/5,2)
+
+  def print_first_day(self):
+
+    print('\nSTOCK\t\t\t\t\tINITIALS\tPRICE/SHARE($)')
+    for stock, data in self.data.items():
+      if stock != 'LBJ':
+        print('{}\t\t\t{}\t\t{}'.format(data['Name'], stock, data['Price']))
+      else:
+        print('{}\t\t{}\t\t{}'.format(data['Name'], stock, data['Price']))
+
+    self.print_exchange_average()
+    self.print_assets()
+
+  def take_inputs(self):
+
+    print('\nWHAT IS YOUR TRANSACTION IN')
+    flag = False
+    while flag != True:
+      new_holdings = []
+      for stock in self.data.keys():
+        try:
+          new_holdings.append(int(input('{}? '.format(stock))))
+        except:
+          print('\nINVALID ENTRY, TRY AGAIN\n')
+          break
+      if len(new_holdings) == 5:
+        flag = self._check_transaction(new_holdings)
+      
+    return new_holdings
+
+  def print_trading_day(self):
+
+    print("STOCK\tPRICE/SHARE\tHOLDINGS\tNET. Value\tPRICE CHANGE")
+    for stock, data, change in zip(self.data.keys(), self.data.values(),self.changes):
+      value = data['Price'] * data['Holdings']
+      print('{}\t{}\t\t{}\t\t{:.2f}\t\t{}'.format(stock, data['Price'], data['Holdings'], value, change))
+
+  def update_cash_assets(self, new_holdings):
+
+    sell=0
+    buy=0
+    for stock, holding in zip(self.data.values(), new_holdings):
+      if holding > 0:
+        buy += stock['Price']*holding
+      
+      elif holding < 0:
+        sell += stock['Price']*abs(holding)
+
+    self.cash_assets = self.cash_assets + sell - buy
+
+  def update_stock_assets(self):
+
+    sum=0
+    for data in self.data.values():
+      sum += data['Price']*data['Holdings']
+
+    self.stock_assets = round(sum,2)
+
+  def print_assets(self):
+
+    print('\nTOTAL STOCK ASSETS ARE: ${:.2f}'.format(self.stock_assets))
+    print('TOTAL CASH ASSETS ARE: ${:.2f}'.format(self.cash_assets))
+    print('TOTAL ASSETS ARE: ${:.2f}'.format(self.total_assets()))
+
+  def _check_transaction(self, new_holdings):
+
+    sum = 0
+    for stock, holding in zip(self.data.values(), new_holdings):
+      if holding > 0:
+        sum += stock['Price']*holding
+      
+      elif holding < 0:
+        if abs(holding) > stock['Holdings']:
+          print('\nYOU HAVE OVERSOLD SOME STOCKS, TRY AGAIN\n')
+          return False
+
+    if sum > self.cash_assets:
+      print('\nYOU HAVE USED ${:.2f} MORE THAN YOU HAVE, TRY AGAIN\n'.format(sum - self.cash_assets))
+      return False
+
+    return True
+
+  def update_holdings(self, new_holdings):
+
+    for stock, new_holding in zip(self.data.values(), new_holdings):
+      stock['Holdings'] += new_holding
+
+def print_instruction():
+
+  print('''
+THIS PROGRAM PLAYS THE STOCK MARKET.  YOU WILL BE GIVEN
+$10,000 AND MAY BUY OR SELL STOCKS.  THE STOCK PRICES WILL
+BE GENERATED RANDOMLY AND THEREFORE THIS MODEL DOES NOT
+REPRESENT EXACTLY WHAT HAPPENS ON THE EXCHANGE.  A TABLE
+OF AVAILABLE STOCKS, THEIR PRICES, AND THE NUMBER OF SHARES
+IN YOUR PORTFOLIO WILL BE PRINTED.  FOLLOWING THIS, THE
+INITIALS OF EACH STOCK WILL BE PRINTED WITH A QUESTION
+MARK.  HERE YOU INDICATE A TRANSACTION.  TO BUY A STOCK
+TYPE +NNN, TO SELL A STOCK TYPE -NNN, WHERE NNN IS THE
+NUMBER OF SHARES.  A BROKERAGE FEE OF 1% WILL BE CHARGED
+ON ALL TRANSACTIONS.  NOTE THAT IF A STOCK'S VALUE DROPS
+TO ZERO IT MAY REBOUND TO A POSITIVE VALUE AGAIN.  YOU
+HAVE $10,000 TO INVEST.  USE INTEGERS FOR ALL YOUR INPUTS.
+(NOTE:  TO GET A 'FEEL' FOR THE MARKET RUN FOR AT LEAST
+10 DAYS)
+          ------------GOOD LUCK!------------\n
+    ''')
+  
+if __name__ == "__main__":
+  
+  print('\t\t      STOCK MARKET')
+  help = input('\nDO YOU WANT INSTRUCTIONS(YES OR NO)? ')
+
+  #Printing Instruction
+  if help == 'YES' or help == 'yes' or help == 'Yes':
+    print_instruction()
+  
+  #Initialize Game
+  Game = Stock_Market()
+
+  #Do first day
+  Game.print_first_day()
+  new_holdings = Game.take_inputs()
+  Game.update_holdings(new_holdings)
+  Game.update_cash_assets(new_holdings)
+  print('\n------------END OF TRADING DAY--------------\n')
+
+  response = 1
+  while response == 1:
+
+    #Simulate a DAY
+    Game.update_prices()
+    Game.print_trading_day()
+    Game.print_exchange_average()
+    Game.update_stock_assets()
+    Game.print_assets()
+
+    response = int(input('\nDO YOU WISH TO CONTINUE (YES-TYPE 1, NO-TYPE 0)? '))
+    if response == 0:
+      break
+    
+    new_holdings = Game.take_inputs()
+    Game.update_holdings(new_holdings)
+    Game.update_cash_assets(new_holdings)
+    print('\n------------END OF TRADING DAY--------------\n')
+
+  print('\nHOPE YOU HAD FUN!!!!')
+  input('')