mirror of
https://github.com/coding-horror/basic-computer-games.git
synced 2025-12-21 23:00:43 -08:00
5caa02958e
I've used an Object-Oriented Approach. The code's a little messy, We'll clean it up later.
209 lines
6.0 KiB
Python
209 lines
6.0 KiB
Python
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('')
|