2016-10-01 04:44:03 +08:00
|
|
|
import os
|
|
|
|
|
import psycopg2
|
|
|
|
|
import subprocess
|
|
|
|
|
|
|
|
|
|
DB_CONN = psycopg2.connect('postgres://{user}:{password}@{host}:{port}/{database}'.format(
|
|
|
|
|
user=os.environ.get('PGUSER', 'postgres'),
|
|
|
|
|
password=os.environ.get('PGPASSWORD', ''),
|
|
|
|
|
host=os.environ.get('PGHOST', 'localhost'),
|
|
|
|
|
port=os.environ.get('PGPORT', '5432'),
|
|
|
|
|
database=os.environ.get('PGDATABASE', 'postgres'),
|
|
|
|
|
))
|
|
|
|
|
CURSOR = DB_CONN.cursor()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def query(q):
|
|
|
|
|
'''
|
|
|
|
|
Query the database.
|
|
|
|
|
'''
|
|
|
|
|
try:
|
|
|
|
|
CURSOR.execute(q)
|
|
|
|
|
return CURSOR
|
|
|
|
|
except:
|
|
|
|
|
DB_CONN.rollback()
|
|
|
|
|
raise
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def commit():
|
|
|
|
|
try:
|
|
|
|
|
DB_CONN.commit()
|
|
|
|
|
except:
|
|
|
|
|
DB_CONN.rollback()
|
|
|
|
|
raise
|
2016-05-12 23:07:21 +08:00
|
|
|
|
|
|
|
|
|
|
|
|
|
def get_tablename_query(column_id, boundary_id, timespan):
|
|
|
|
|
"""
|
|
|
|
|
given a column_id, boundary-id (us.census.tiger.block_group), and
|
|
|
|
|
timespan, give back the current table hash from the data observatory
|
|
|
|
|
"""
|
2016-10-01 04:44:03 +08:00
|
|
|
return """
|
2017-01-10 10:28:38 +08:00
|
|
|
SELECT numer_tablename, numer_geomref_colname, numer_tid,
|
|
|
|
|
geom_tablename, geom_geomref_colname, geom_tid
|
|
|
|
|
FROM observatory.obs_meta
|
|
|
|
|
WHERE numer_id = '{numer_id}' AND
|
|
|
|
|
geom_id = '{geom_id}' AND
|
|
|
|
|
numer_timespan = '{numer_timespan}'
|
|
|
|
|
""".format(numer_id=column_id,
|
|
|
|
|
geom_id=boundary_id,
|
|
|
|
|
numer_timespan=timespan)
|
2016-05-12 23:07:21 +08:00
|
|
|
|
|
|
|
|
|
2016-10-01 04:44:03 +08:00
|
|
|
METADATA_TABLES = ['obs_table', 'obs_column_table', 'obs_column', 'obs_column_tag',
|
2016-10-04 04:36:14 +08:00
|
|
|
'obs_tag', 'obs_column_to_column', 'obs_dump_version', 'obs_meta',
|
|
|
|
|
'obs_meta_numer', 'obs_meta_denom', 'obs_meta_geom',
|
2016-12-13 05:25:59 +08:00
|
|
|
'obs_meta_timespan', 'obs_column_table_tile',
|
|
|
|
|
'obs_column_table_tile_simple']
|
2016-05-12 23:07:21 +08:00
|
|
|
|
2016-10-04 04:36:14 +08:00
|
|
|
FIXTURES = [
|
|
|
|
|
('us.census.acs.B01003001_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B01001002_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B01001026_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B01002001_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B03002003_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B03002004_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B03002006_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B03002012_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B05001006_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B08006001_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B08006002_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
2017-01-10 10:28:38 +08:00
|
|
|
('us.census.acs.B08301010_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
2016-10-04 04:36:14 +08:00
|
|
|
('us.census.acs.B08006009_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B08006011_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B08006015_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B08006017_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B09001001_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B11001001_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B14001001_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B14001002_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B14001005_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B14001006_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B14001007_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B14001008_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B15003001_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B15003017_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B15003022_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B15003023_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B16001001_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B16001002_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B16001003_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B17001001_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B17001002_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19013001_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19083001_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19301001_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B25001001_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B25002003_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B25004002_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B25004004_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B25058001_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B25071001_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B25075001_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B25075025_quantile', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B01003001', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B01001002', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B01001026', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B01002001', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B03002003', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B03002004', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B03002006', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B03002012', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B03002005', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B03002008', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B03002009', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B03002002', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B11001001', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B15003001', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B15003017', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B15003019', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B15003020', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B15003021', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B15003022', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B15003023', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19013001', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19301001', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B25001001', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B25002003', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B25004002', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B25004004', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B25058001', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B25071001', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B25075001', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B25075025', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B25081002', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B08134001', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B08134002', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001002', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001003', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001004', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001005', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001006', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001007', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001008', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001009', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001010', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001011', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001012', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001013', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001014', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001015', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001016', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001017', 'us.census.tiger.block_group', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B01001002', 'us.census.tiger.block_group', '2010 - 2014'),
|
2016-05-13 23:23:10 +08:00
|
|
|
('us.census.acs.B01003001', 'us.census.tiger.census_tract', '2010 - 2014'),
|
2017-01-10 10:28:38 +08:00
|
|
|
('us.census.acs.B01001002', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B01001026', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B01002001', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B03002003', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B03002004', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B03002006', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B03002012', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B03002005', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B03002008', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B03002009', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B03002002', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B11001001', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B15003001', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B15003017', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B15003019', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B15003020', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B15003021', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B15003022', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B15003023', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19013001', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19083001', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19301001', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B25001001', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B25002003', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B25004002', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B25004004', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B25058001', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B25071001', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B25075001', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B25075025', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B25081002', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B08134001', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B08134002', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B08134008', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B08134008', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B08134010', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001002', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001003', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001004', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001005', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001006', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001007', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001008', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001009', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001010', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001011', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001012', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001013', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001014', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001015', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001016', 'us.census.tiger.census_tract', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B19001017', 'us.census.tiger.census_tract', '2010 - 2014'),
|
2016-05-17 20:28:23 +08:00
|
|
|
('us.census.spielman_singleton_segments.X10', 'us.census.tiger.census_tract', '2010 - 2014'),
|
2016-10-04 04:36:14 +08:00
|
|
|
('us.census.spielman_singleton_segments.X55', 'us.census.tiger.census_tract', '2010 - 2014'),
|
2016-05-12 23:07:21 +08:00
|
|
|
('us.zillow.AllHomes_Zhvi', 'us.census.tiger.zcta5', '2014-01'),
|
2016-10-04 04:36:14 +08:00
|
|
|
('us.zillow.AllHomes_Zhvi', 'us.census.tiger.zcta5', '2016-06'),
|
2017-01-10 10:28:38 +08:00
|
|
|
('whosonfirst.wof_country_name', 'whosonfirst.wof_country_geom', '2016'),
|
|
|
|
|
('us.census.acs.B01003001', 'us.census.tiger.zcta5_clipped', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B01003001', 'us.census.tiger.block_group_clipped', '2010 - 2014'),
|
|
|
|
|
('us.census.acs.B01003001', 'us.census.tiger.census_tract_clipped', '2010 - 2014'),
|
|
|
|
|
('us.census.tiger.fullname', 'us.census.tiger.pointlm_geom', '2016'),
|
|
|
|
|
('us.census.tiger.fullname', 'us.census.tiger.prisecroads_geom', '2016'),
|
|
|
|
|
('us.census.tiger.name', 'us.census.tiger.county', '2015'),
|
2016-05-12 23:07:21 +08:00
|
|
|
]
|
|
|
|
|
|
2016-10-04 04:36:14 +08:00
|
|
|
OUTFILE_PATH = os.path.join(os.path.dirname(__file__), '..',
|
|
|
|
|
'src/pg/test/fixtures/load_fixtures.sql')
|
|
|
|
|
DROPFILE_PATH = os.path.join(os.path.dirname(__file__), '..',
|
|
|
|
|
'src/pg/test/fixtures/drop_fixtures.sql')
|
2016-10-01 04:44:03 +08:00
|
|
|
|
|
|
|
|
def dump(cols, tablename, where=''):
|
|
|
|
|
|
|
|
|
|
with open(DROPFILE_PATH, 'a') as dropfile:
|
|
|
|
|
dropfile.write('DROP TABLE IF EXISTS observatory.{tablename};\n'.format(
|
|
|
|
|
tablename=tablename,
|
|
|
|
|
))
|
|
|
|
|
|
|
|
|
|
subprocess.check_call('pg_dump -x --section=pre-data -t observatory.{tablename} '
|
2016-10-04 04:36:14 +08:00
|
|
|
' | sed "s:SET search_path.*::" '
|
2016-10-01 04:44:03 +08:00
|
|
|
' | sed "s:CREATE TABLE :CREATE TABLE observatory.:" '
|
|
|
|
|
' | sed "s:ALTER TABLE.*OWNER.*::" '
|
2016-12-22 05:53:53 +08:00
|
|
|
' | sed "s:SET idle_in_transaction_session_timeout.*::" '
|
2016-10-01 04:44:03 +08:00
|
|
|
' >> {outfile}'.format(
|
|
|
|
|
tablename=tablename,
|
|
|
|
|
outfile=OUTFILE_PATH,
|
|
|
|
|
), shell=True)
|
|
|
|
|
|
|
|
|
|
with open(OUTFILE_PATH, 'a') as outfile:
|
|
|
|
|
outfile.write('COPY observatory."{}" FROM stdin WITH CSV HEADER;\n'.format(tablename))
|
|
|
|
|
|
|
|
|
|
subprocess.check_call('''
|
|
|
|
|
psql -c "COPY (SELECT {cols} \
|
|
|
|
|
FROM observatory.{tablename} {where}) \
|
|
|
|
|
TO STDOUT WITH CSV HEADER" >> {outfile}'''.format(
|
|
|
|
|
cols=cols,
|
|
|
|
|
tablename=tablename,
|
|
|
|
|
where=where,
|
|
|
|
|
outfile=OUTFILE_PATH,
|
|
|
|
|
), shell=True)
|
|
|
|
|
|
|
|
|
|
with open(OUTFILE_PATH, 'a') as outfile:
|
|
|
|
|
outfile.write('\\.\n\n')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def main():
|
|
|
|
|
unique_tables = set()
|
|
|
|
|
|
2016-10-04 04:36:14 +08:00
|
|
|
for f in FIXTURES:
|
2016-10-01 04:44:03 +08:00
|
|
|
column_id, boundary_id, timespan = f
|
|
|
|
|
tablename_query = get_tablename_query(column_id, boundary_id, timespan)
|
2016-10-04 04:36:14 +08:00
|
|
|
resp = query(tablename_query).fetchone()
|
|
|
|
|
if resp:
|
2017-01-10 10:28:38 +08:00
|
|
|
numer_tablename, numer_colname, numer_table_id = resp[0:3]
|
|
|
|
|
geom_tablename, geom_colname, geom_table_id = resp[3:6]
|
2016-10-04 04:36:14 +08:00
|
|
|
else:
|
2017-01-10 10:28:38 +08:00
|
|
|
raise Exception("Could not find table for {}, {}, {}".format(
|
2016-10-04 04:36:14 +08:00
|
|
|
column_id, boundary_id, timespan))
|
2017-01-10 10:28:38 +08:00
|
|
|
numer = (numer_tablename, numer_colname, numer_table_id, )
|
|
|
|
|
geom = (geom_tablename, geom_colname, geom_table_id, )
|
|
|
|
|
if numer not in unique_tables:
|
|
|
|
|
print(numer)
|
|
|
|
|
unique_tables.add(numer)
|
|
|
|
|
if geom not in unique_tables:
|
|
|
|
|
print(geom)
|
|
|
|
|
unique_tables.add(geom)
|
2016-10-01 04:44:03 +08:00
|
|
|
|
|
|
|
|
print unique_tables
|
|
|
|
|
|
|
|
|
|
with open(OUTFILE_PATH, 'w') as outfile:
|
|
|
|
|
outfile.write('SET client_min_messages TO WARNING;\n\\set ECHO none\n')
|
|
|
|
|
outfile.write('CREATE SCHEMA IF NOT EXISTS observatory;\n\n')
|
|
|
|
|
|
|
|
|
|
with open(DROPFILE_PATH, 'w') as dropfile:
|
|
|
|
|
dropfile.write('SET client_min_messages TO WARNING;\n\\set ECHO none\n')
|
|
|
|
|
|
|
|
|
|
for tablename in METADATA_TABLES:
|
|
|
|
|
print(tablename)
|
|
|
|
|
if tablename == 'obs_meta':
|
|
|
|
|
where = "WHERE " + " OR ".join([
|
|
|
|
|
"(numer_id, geom_id, numer_timespan) = ('{}', '{}', '{}')".format(
|
|
|
|
|
numer_id, geom_id, timespan)
|
2016-10-04 04:36:14 +08:00
|
|
|
for numer_id, geom_id, timespan in FIXTURES
|
2016-10-01 04:44:03 +08:00
|
|
|
])
|
|
|
|
|
elif tablename == 'obs_meta_numer':
|
|
|
|
|
where = "WHERE " + " OR ".join([
|
|
|
|
|
"numer_id IN ('{}', '{}')".format(numer_id, geom_id)
|
2016-10-04 04:36:14 +08:00
|
|
|
for numer_id, geom_id, timespan in FIXTURES
|
2016-10-01 04:44:03 +08:00
|
|
|
])
|
|
|
|
|
elif tablename == 'obs_meta_denom':
|
|
|
|
|
where = "WHERE " + " OR ".join([
|
|
|
|
|
"denom_id IN ('{}', '{}')".format(numer_id, geom_id)
|
2016-10-04 04:36:14 +08:00
|
|
|
for numer_id, geom_id, timespan in FIXTURES
|
2016-10-01 04:44:03 +08:00
|
|
|
])
|
|
|
|
|
elif tablename == 'obs_meta_geom':
|
|
|
|
|
where = "WHERE " + " OR ".join([
|
|
|
|
|
"geom_id IN ('{}', '{}')".format(numer_id, geom_id)
|
2016-10-04 04:36:14 +08:00
|
|
|
for numer_id, geom_id, timespan in FIXTURES
|
2016-10-01 04:44:03 +08:00
|
|
|
])
|
|
|
|
|
elif tablename == 'obs_meta_timespan':
|
|
|
|
|
where = "WHERE " + " OR ".join([
|
|
|
|
|
"timespan_id = ('{}')".format(timespan)
|
2016-10-04 04:36:14 +08:00
|
|
|
for numer_id, geom_id, timespan in FIXTURES
|
2016-10-01 04:44:03 +08:00
|
|
|
])
|
|
|
|
|
elif tablename == 'obs_column':
|
|
|
|
|
where = "WHERE " + " OR ".join([
|
|
|
|
|
"id IN ('{}', '{}')".format(numer_id, geom_id)
|
2016-10-04 04:36:14 +08:00
|
|
|
for numer_id, geom_id, timespan in FIXTURES
|
2016-10-01 04:44:03 +08:00
|
|
|
])
|
2016-10-04 04:36:14 +08:00
|
|
|
elif tablename == 'obs_column_tag':
|
2016-10-01 04:44:03 +08:00
|
|
|
where = "WHERE " + " OR ".join([
|
|
|
|
|
"column_id IN ('{}', '{}')".format(numer_id, geom_id)
|
2016-10-04 04:36:14 +08:00
|
|
|
for numer_id, geom_id, timespan in FIXTURES
|
2016-10-01 04:44:03 +08:00
|
|
|
])
|
2016-12-13 05:25:59 +08:00
|
|
|
elif tablename in ('obs_column_table', 'obs_column_table_tile',
|
|
|
|
|
'obs_column_table_tile_simple'):
|
2017-01-10 10:28:38 +08:00
|
|
|
where = '''WHERE table_id IN ({table_ids}) AND
|
|
|
|
|
(column_id IN ({numer_ids}) OR column_id IN ({geom_ids}))
|
|
|
|
|
'''.format(
|
2016-10-04 04:36:14 +08:00
|
|
|
numer_ids=','.join(["'{}'".format(x) for x, _, _ in FIXTURES]),
|
|
|
|
|
geom_ids=','.join(["'{}'".format(x) for _, x, _ in FIXTURES]),
|
2017-01-10 10:28:38 +08:00
|
|
|
table_ids=','.join(["'{}'".format(x) for _, _, x in unique_tables])
|
2016-10-04 04:36:14 +08:00
|
|
|
)
|
2016-10-01 04:44:03 +08:00
|
|
|
elif tablename == 'obs_column_to_column':
|
|
|
|
|
where = "WHERE " + " OR ".join([
|
|
|
|
|
"source_id IN ('{}', '{}') OR target_id IN ('{}', '{}')".format(
|
|
|
|
|
numer_id, geom_id, numer_id, geom_id)
|
2016-10-04 04:36:14 +08:00
|
|
|
for numer_id, geom_id, timespan in FIXTURES
|
2016-10-01 04:44:03 +08:00
|
|
|
])
|
|
|
|
|
elif tablename == 'obs_table':
|
2016-10-04 04:36:14 +08:00
|
|
|
where = 'WHERE timespan IN ({timespans}) ' \
|
|
|
|
|
'OR id IN ({table_ids}) '.format(
|
|
|
|
|
timespans=','.join(["'{}'".format(x) for _, _, x in FIXTURES]),
|
2017-01-10 10:28:38 +08:00
|
|
|
table_ids=','.join(["'{}'".format(x) for _, _, x in unique_tables])
|
2016-10-04 04:36:14 +08:00
|
|
|
)
|
2016-10-01 04:44:03 +08:00
|
|
|
else:
|
|
|
|
|
where = ''
|
|
|
|
|
dump('*', tablename, where)
|
|
|
|
|
|
2017-01-10 10:28:38 +08:00
|
|
|
for tablename, colname, table_id in unique_tables:
|
|
|
|
|
if 'zcta5' in table_id or 'zillow_zip' in table_id:
|
2016-10-01 04:44:03 +08:00
|
|
|
where = '\'11%\''
|
|
|
|
|
compare = 'LIKE'
|
2017-01-10 10:28:38 +08:00
|
|
|
elif 'pri_sec_roads' in table_id or 'point_landmark' in table_id:
|
2016-12-30 01:07:38 +08:00
|
|
|
dump('*', tablename, 'WHERE geom && ST_MakeEnvelope(-74,40.69,-73.9,40.72, 4326)')
|
|
|
|
|
continue
|
2017-01-10 10:28:38 +08:00
|
|
|
elif 'whosonfirst' in table_id:
|
2016-10-01 04:44:03 +08:00
|
|
|
where = '(\'85632785\',\'85633051\',\'85633111\',\'85633147\',\'85633253\',\'85633267\')'
|
|
|
|
|
compare = 'IN'
|
|
|
|
|
else:
|
|
|
|
|
where = '\'36047%\''
|
|
|
|
|
compare = 'LIKE'
|
|
|
|
|
print ' '.join(['*', tablename, "WHERE {}::text {} {}".format(colname, compare, where)])
|
|
|
|
|
dump('*', tablename, "WHERE {}::text {} {}".format(colname, compare, where))
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
|
main()
|
