hackdisrupt.py

# -*- coding: utf-8 -*-
from flask import Flask, jsonify, Response
import requests
import os
from ConcurOAuthPython import getbasic, getTokenGivenUsernamePasswordAndConsumerKey
from BeautifulSoup import BeautifulSoup
from datetime import datetime, timedelta
import hashlib
import binascii
import time
from evernote.edam.notestore.ttypes import NoteFilter
import evernote.edam.type.ttypes as Types
from evernote.api.client import EvernoteClient

# Sandbox: https://sandbox.evernote.com/api/DeveloperToken.action
# Production: https://www.evernote.com/api/DeveloperToken.action
auth_token = "S=s1:U=8f65c:E=14fa7113473:C=1484f600570:P=1cd:A=en-devtoken:V=2:H=484744fd0ffa906797416ae02ce5cd9c"
#Constants
USER = "user46@concurdisrupt.com"
PWD = "disrupt"
KEY = 'mbGLfEhU1rUUG8IOImdmVt'
#TOKEN = 'NPuB/Wr+a9OkYNWz9Mw/kWNWvyk='
SAMPLE_REPORT_NAME = 'Disrupt Hackathon NYC Trip'
SAMPLE_REPORT_ID = 'D7E30A5A1E1F4ACF82FE'
city_name = "NYC"
TripID = 'nHyQT$pcgFtPlchwTlyhTG$pjyel9xI$sVJ9OTTrUw'
SITE_ROOT = os.path.realpath(os.path.dirname(__file__))

#Initializing FLASK app
app = Flask(__name__)

#Getting the Current TOKEN
TOKEN = getTokenGivenUsernamePasswordAndConsumerKey('user46@concurdisrupt.com', 'disrupt', KEY)['Access_Token']['Token']

#oauth
auth = {
    'Authorization': "OAuth %s" %  TOKEN,
    'Accept':'application/xml',
    'lastModifiedDate': '2001-01-01' #This will get older itinerariesk
}
client = EvernoteClient(token=auth_token, sandbox=True)

note_store = client.get_note_store()

# Listar todos notebooks:
notebooks = note_store.listNotebooks()
# print "Achei ", len(notebooks), " notebooks:"
# for notebook in notebooks:
# print "  * ", notebook.name

# Criar uma nova nota:
# print "\nCriando uma nova nota no notebook principal\n"

note = Types.Note()
note.title = "Business Travel to %s" % (city_name)

# Anexando uma imagem:
image = open( os.path.join(SITE_ROOT, 'NY.png'), 'rb').read()

md5 = hashlib.md5()
md5.update(image)
hash = md5.digest()

data = Types.Data()
data.size = len(image)
data.bodyHash = hash
data.body = image

resource = Types.Resource()
resource.mime = 'image/png'
resource.data = data

# Adicionando o novo Resource na lista de resources dessa nota
note.resources = [resource]

# Para exibir a imagem no meio da nota, soh precisamos do hash MD5 dela
hash_hex = binascii.hexlify(hash)

# ENML = Evernote Markup Language. It's an HTML subset with some extra tags
# note.content = '<?xml version="1.0" encoding="UTF-8"?>'
# note.content += '<!DOCTYPE en-note SYSTEM "http://xml.evernote.com/pub/enml2.dtd">'
# note.content += '<en-note>'
# note.content += 'Esta eh uma nova nota, inserida direto no notebook principal :)<br/>'
# note.content += 'Olha aqui o logo do Evernote:<br/>'
# note.content += '<en-media type="image/png" hash="' + hash_hex + '"/>'
# note.content += '</en-note>'

flight_bool, history_bool, taxi_bool, hotel_bool, weather_bool, tip_bool = True, True, True, True, True, True

@app.route('/')
def home():
    pass

@app.route('/itineraries/')
def itineraries():
    """
     Returns all itineraries
    """
    resp = requests.get('https://www.concursolutions.com/api/travel/trip/v1.1',headers = auth)
    return Response(resp.text, mimetype='application/xml')

@app.route('/itinerary/info')
def itinerary_info(TripID = 'nHyQT$pcgFtPlchwnqtoRjd1MgMLqRPunn4pQShg'):
    """
     Returns all itineraries
    """
    TripID = 'nHyQT$pcgFtPlchwTlyhTG$pjyel9xI$sVJ9OTTrUw'
    resp = requests.get('https://www.concursolutions.com/api/travel/trip/v1.1/%s' % TripID,headers = auth)
    root = BeautifulSoup(resp.text)
    note=""
    for booking in root.bookings.findAll('booking'):
        for air in booking.findAll('air'):
            note += """
        <div>
            <ul><li><strong>Flight</strong></li></ul>
        </div>
        <div>%s -&gt; %s</div>
        <div><en-todo/> arrive on %s by %s</div>
        <div><en-todo/> takeoff is at %s</div>
        <hr></hr>
            """ % (
                air.startcitycode.text,
                air.endcitycode.text,
                air.startcitycode.text,
                (datetime.strptime(air.startdatelocal.text, '%Y-%m-%dT%H:%M:%S') - timedelta(hours=3)).strftime("%m/%d - %H:%M"),
                datetime.strptime(air.startdatelocal.text, '%Y-%m-%dT%H:%M:%S').strftime("%H:%M"),
                )
    note += """
        <div>
            <ul><li><strong>Hotel</strong></li></ul>
        </div>
        <div>You're staying at:</div>
        <div><a href="https://www.google.com/maps/preview?q=Times+Square+Hilton+New+York">%s</div>
        <hr></hr>
            """ % (
                root.bookings.findAll('booking')[0].hotel.find('name').text,
                )
    #Counting number of Taxi Cabs
    resp = requests.get('https://www.concursolutions.com/api/v3.0/expense/entries/',headers = auth)
    root = BeautifulSoup(resp.text)
    cont_cabs = 0
    for t in root.findAll('expensetypecode'):
        if t.text == 'TAXIX':
            cont_cabs = cont_cabs + 1
    note += """
        <div>
            <ul><li><strong>Taxi</strong></li></ul>
        </div>
        <div>The last time you've been there you used %s taxi cabs.</div>
        <div><en-todo/> Download <a href="http://www.uber.com/">Uber</a> (it works the best in %s)</div>
        <hr></hr>
        """% (
        cont_cabs,
        'NYC',
    )
    return note
    #return Response(resp.text, mimetype='application/xml')

@app.route('/itinerary/all')
def all_itinerary_info():
    resp = requests.get('https://www.concursolutions.com/api/travel/trip/v1.1',headers = auth)
    root = BeautifulSoup(resp.text)
    note = ""
    for i in root.findAll("itineraryinfo"):
        note += itinerary_info(i.tripid.text)

    return note

@app.route('/entries/')
def entries():
    """
     Returns all expense entries
    """
    #Expense Entries
    resp = requests.get('https://www.concursolutions.com/api/v3.0/expense/entries/',headers = auth)
    return Response(resp.text, mimetype='application/xml')

@app.route('/integrate/')
def integrate():
    note.content = """<?xml version="1.0" encoding="UTF-8"?>
    <!DOCTYPE en-note SYSTEM "http://xml.evernote.com/pub/enml2.dtd">
    <en-note>
        <div>
            """
    if flight_bool:
        resp = requests.get('https://www.concursolutions.com/api/travel/trip/v1.1/%s' % TripID,headers = auth)
        root = BeautifulSoup(resp.text)
        for booking in root.bookings.findAll('booking'):
            for air in booking.findAll('air'):
                note.content += """
            <div>
                <ul><li><strong>Flight</strong></li></ul>
            </div>
            <div>%s -&gt; %s</div>
            <div><en-todo/> arrive on %s by %s</div>
            <div><en-todo/> takeoff is at %s</div>
            <hr></hr>
                """ % (
                    air.startcitycode.text,
                    air.endcitycode.text,
                    air.startcitycode.text,
                    (datetime.strptime(air.startdatelocal.text, '%Y-%m-%dT%H:%M:%S') - timedelta(hours=3)).strftime("%m/%d - %H:%M"),
                    datetime.strptime(air.startdatelocal.text, '%Y-%m-%dT%H:%M:%S').strftime("%H:%M"),
                    )
    if hotel_bool:
        note.content += """
            <div>
                <ul><li><strong>Hotel</strong></li></ul>
            </div>
            <div>You're staying at:</div>
            <div>%s</div>
            <hr></hr>
                """ % (
            root.bookings.findAll('booking')[0].hotel.find('name').text,
            )

    if taxi_bool:
        #Counting number of Taxi Cabs
        resp = requests.get('https://www.concursolutions.com/api/v3.0/expense/entries/',headers = auth)
        root = BeautifulSoup(resp.text)
        cont_cabs = 0
        for t in root.findAll('expensetypecode'):
            if t.text == 'TAXIX':
                cont_cabs = cont_cabs + 1
        note.content += """
            <div>
                <ul><li><strong>Taxi</strong></li></ul>
            </div>
            <div>The last time you've been there you used %s taxi cabs.</div>
            <div><en-todo/> Download <a href="http://www.uber.com/">Uber</a> (it works the best in %s)</div>
            <hr></hr>
            """% (
            cont_cabs,
            'NYC',
            )

    if weather_bool:
        note.content += """
            <div>
                <ul><li><strong>Weather</strong></li></ul>
            </div>
            <div>It'll be summertime in """+ city_name +"""</div>
            <div><en-todo/> Grab a coat, it'll be rainy</div>
            <hr></hr>
            """

    def get_tip_notes(start_date, end_date):
        tip_notes = []

        noteFilter = NoteFilter()
        noteFilter.words = "tag:tip created:%s -created:%s" % (start_date, end_date) # notes with tag #tip created between 2012-01-01 and 2014-09-08 (flight dates)
        note_list = note_store.findNotes(auth_token, noteFilter, 0, 10)

        for note in note_list.notes:
            guid = note.guid
            title = note.title
            user_id = "587356"
            shard_id = "s1"
            url = "evernote:///view/%s/%s/%s/%s/" % (user_id, shard_id, guid, guid)
            tip_notes.append( '<div><en-todo/> %s (<a href="%s">view full note</a>)</div>' % (title, url) )
        return tip_notes

    if tip_bool:
        note.content += """
            <div>
                <ul><li><strong>#tip</strong></li></ul>
            </div>
            <div>When you last visited there, you wrote the following #tip:</div>"""
        tip_notes = get_tip_notes("20120101", "20140908")
        for tip_note in tip_notes:
            note.content += tip_note
        note.content += """
            <hr></hr>
            """

    note.content += """
        </div>
        """
    note.content += """
        <div>
            <br clear="none"/>
            <en-media type="image/png" hash="%s"/>
        </div>
    </en-note>""" % (hash_hex)

    # Set our beloved bizmem tag :)
    note.tagNames = ["bizmem"]

    # Set the reminder
    # for demo purposes, we'll ignore the actual useful reminder date (two days before the flight) and set it to today
    now = int(round(time.time() * 1000))
    then = now + 60*60*1000 # one hour after `now`

    # init NoteAttributes instance
    note.attributes = Types.NoteAttributes()
    note.attributes.reminderOrder = now
    note.attributes.reminderTime = then
    # Finally, save the note
    created_note = note_store.createNote(note)
    return "Note created successfully! GUID: %s" % (created_note.guid)

if __name__ == '__main__':
    app.run(debug = True)