#!/usr/bin/env python
# encoding: utf-8
"""
cdb_change_plot.py

Created by Nicholas Tatonetti on 2011-01-30.
Copyright (c) 2011 Stanford University. All rights reserved.
"""

import os
import csv
import sys
import MySQLdb

if __name__ == '__main__':
    
    db = MySQLdb.connect(host="localhost", port=3307, user="root", passwd="enter_your_password",db="project_aers")
    c = db.cursor()
    
    query = """
    select stitch_id, name, e.prr, p.prr, p.prr - e.prr as prr_diff
    from pred_drug_events_e5 e 
    join prop_pred_drug_events p using (stitch_id, umls_id)
    join drugs using (stitch_id)
    join
    (
    	select distinct stitch_id
    	from clinical_stanford.med_class
    	where description = 'Antiarrhythmics'
    ) dm_drugs using (stitch_id)
    where umls_id = 'C0003811' # arrythmia
    order by p.prr desc;
    """
    print c.execute(query)
    
    data = c.fetchall()
    
    e_prr_file = open(os.path.expanduser('~/Dropbox/e_prr_scores.csv'), 'w')
    e_prr_writer = csv.writer(e_prr_file)
    
    p_prr_file = open(os.path.expanduser('~/Dropbox/p_prr_scores.csv'), 'w')
    p_prr_writer = csv.writer(p_prr_file)
    
    for i,(sid, name, ep, pp, pdiff) in enumerate(data):
        e_prr_writer.writerow([ep, i])
        p_prr_writer.writerow([pp, i])
        
        drug_file = open(os.path.expanduser('~/Dropbox/%s_prr_line.csv' % name), 'w')
        drug_file_writer = csv.writer(drug_file)
        drug_file_writer.writerow([ep,i])
        drug_file_writer.writerow([pp,i])
        drug_file.close()
    
    e_prr_file.close()
    p_prr_file.close()