In [2]:

    

import sys
sys.path.append('../')
#print sys.path


import disaggregator.PecanStreetDatasetAdapter as pecan
import disaggregator.appliance


    

In [3]:

    

import disaggregator
disaggregator = reload(disaggregator)
from disaggregator import PecanStreetDatasetAdapter as pecan


    

In [4]:

    

user_name = 'USERNAME'
pw='PASSWORD'
host = "db.wiki-energy.org"
port = "5432"
db = "postgres"
db_url = "postgresql"+"://"+user_name+":"+pw+"@"+host+":"+port+"/"+db


    

In [5]:

    

p = pecan(db_url)
tables= p.get_table_names('shared')
[i,a] = p.get_table_metadata('shared',str(tables[0]))


    

In [7]:

    

schema = '\"PecanStreet_SharedData\"'
query = 'select * from {0}.{1} where dataid={2}'.format(schema,tables[0],i[0])
df = p.get_dataframe(query).fillna(0)


    

In [8]:

    

temp = p.clean_dataframe(df,'shared',[])


    

    




Index([u'dataid', u'time', u'use', u'air1', u'air2', u'air3', u'airwindowunit1', u'aquarium1', u'bathroom1', u'bathroom2', u'bedroom1', u'bedroom2', u'bedroom3', u'bedroom4', u'bedroom5', u'car1', u'clotheswasher1', u'clotheswasher_dryg1', u'diningroom1', u'diningroom2', u'dishwasher1', u'disposal1', u'drye1', u'dryg1', u'freezer1', u'furnace1', u'furnace2', u'garage1', u'garage2', u'gen', u'grid', u'heater1', u'housefan1', u'icemaker1', u'jacuzzi1', u'kitchen1', u'kitchen2', u'kitchenapp1', u'kitchenapp2', u'lights_plugs1', u'lights_plugs2', u'lights_plugs3', u'lights_plugs4', u'lights_plugs5', u'lights_plugs6', u'livingroom1', u'livingroom2', u'microwave1', u'office1', u'outsidelights_plugs1', u'outsidelights_plugs2', u'oven1', u'oven2', u'pool1', u'pool2', u'poollight1', u'poolpump1', u'pump1', u'range1', u'refrigerator1', u'refrigerator2', u'security1', u'shed1', u'sprinkler1', u'unknown1', u'unknown2', u'unknown3', u'unknown4', u'utilityroom1', u'venthood1', u'waterheater1', u'waterheater2', u'winecooler1'], dtype='object')

In [9]:

    

test = p.get_month_traces_per_dataid('shared',tables[0],i[0])


    

    




Index([u'dataid', u'time', u'use', u'air1', u'air2', u'air3', u'airwindowunit1', u'aquarium1', u'bathroom1', u'bathroom2', u'bedroom1', u'bedroom2', u'bedroom3', u'bedroom4', u'bedroom5', u'car1', u'clotheswasher1', u'clotheswasher_dryg1', u'diningroom1', u'diningroom2', u'dishwasher1', u'disposal1', u'drye1', u'dryg1', u'freezer1', u'furnace1', u'furnace2', u'garage1', u'garage2', u'gen', u'grid', u'heater1', u'housefan1', u'icemaker1', u'jacuzzi1', u'kitchen1', u'kitchen2', u'kitchenapp1', u'kitchenapp2', u'lights_plugs1', u'lights_plugs2', u'lights_plugs3', u'lights_plugs4', u'lights_plugs5', u'lights_plugs6', u'livingroom1', u'livingroom2', u'microwave1', u'office1', u'outsidelights_plugs1', u'outsidelights_plugs2', u'oven1', u'oven2', u'pool1', u'pool2', u'poollight1', u'poolpump1', u'pump1', u'range1', u'refrigerator1', u'refrigerator2', u'security1', u'shed1', u'sprinkler1', u'unknown1', u'unknown2', u'unknown3', u'unknown4', u'utilityroom1', u'venthood1', u'waterheater1', u'waterheater2', u'winecooler1'], dtype='object')

In [12]:

    

pair = []
pair.append(test[0])
pair.append(test[1])


    

In [13]:

    

p.verify_same_range(pair)


    

    




---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-13-151a20b4f7ba> in <module>()
----> 1 p.verify_same_range(pair)

/Users/sabina/wikienergy/disaggregator/PecanStreetDatasetAdapter.py in verify_same_range(self, pair)
     48         #check that start and end times are the same
     49         #check that
---> 50             start_time_one = pair[0].series.index(pair[0].series[0])
     51             start_time_two = pair[0].series.index(pair[0].series[0])
     52             print start_time_one == start_time_two

TypeError: 'DatetimeIndex' object is not callable

In [13]:

    

import pickle


    

In [16]:

    

pickle.dump(test, open('pickle_test.p','wb'))


    

In [11]:

    




    

    
Out[11]:





40

In [ ]: