def make_trade():
stock = input("What stock do you want to trade? ").upper()
quantity = int(input("How many shares are you trading? "))
direction = input("Are you [b]uying or [s]elling? (b/s) ")
if direction not in ('b', 's'):
return "Direction must be 'b' or 's'"
else:
direction = 'buy' if direction == 'b' else 'sell'
traded_price = int(input("What is price per share in the trade in whole pennies? "))
record_trade(stock, quantity=quantity, direction=direction, traded_price=traded_price)
return "Trade recorded"
def setUp(self):
self.common_stock = stock_factory(
stock_type='common',
stock_symbol='Tea',
last_dividend=7,
par_value=100
)
self.preferred_stock = stock_factory(
stock_type='preferred',
stock_symbol='Gin',
last_dividend=8,
fixed_dividend=0.02,
par_value=100
)
for count in range(15):
for stock in [self.common_stock, self.preferred_stock]:
record_trade(
stock.stock_symbol,
time=datetime.datetime.now() - datetime.timedelta(minutes=count),
quantity=(count+1),
direction='buy',
traded_price=10
)