#!/usr/bin/python2.7

import matplotlib.pyplot as plt
import numpy as np
import urllib
from matplotlib.dates import strpdate2num

def graph_data(stock):
    # clump same logic together
    # part 1
    fig = plt.figure()
    ax1 = plt.subplot2grid([1, 1], [0, 0])

    # part 2
    stock_price_url = 'http://chartapi.finance.yahoo.com/instrument/1.0/'\
        +stock+ '/chartdata;type=quote;range=10y/csv'

    source_code = urllib.urlopen(stock_price_url).read().decode()
    stock_data = []
    split_source = source_code.split('\n')

    for line in split_source:
        split_line = line.split(',')
        if len(split_line) == 6:
            if 'values' not in line:
                stock_data.append(line)

    date, closep, highp, lowp, openp, volume = np.loadtxt(stock_data,
            delimiter=',', unpack=True,
            converters={0:strpdate2num('%Y%m%d')})

    ax1.plot([], [], linewidth=5, label='loss', color='r', alpha=0.5)
    ax1.plot([], [], linewidth=5, label='gain', color='g', alpha=0.5)

    for label in ax1.xaxis.get_ticklabels():
        label.set_rotation(45)

    ax1.plot_date(date, closep, '-', label='price', linewidth=2)
    ax1.fill_between(date, closep, closep[0], where=(closep <= closep[0]),
            facecolor='g', alpha=0.3)
    ax1.fill_between(date, closep, closep[0], where=(closep > closep[0]),
            facecolor='r', alpha=0.3)
    ax1.axhline(closep[0], color='k', linewidth=5)
    ax1.grid(True)

    ax1.set_yticks([0, 10, 20, 30, 40])

    ax1.spines['left'].set_color('c')
    ax1.spines['right'].set_color('r')
    ax1.spines['top'].set_visible(False)

    ax1.spines['left'].set_linewidth(5)
    ax1.tick_params(axis='x', colors='#f06215')
    # part 3
    plt.xlabel('date')
    plt.ylabel('price')
    plt.title('Interesting Graph')
    plt.legend()
    plt.subplots_adjust(left=0.09, bottom=0.18, right=0.94, top=0.93,
            wspace=0.2, hspace=0)
    plt.show()

graph_data('EBAY')

