#!/usr/bin/env python

"""
Download any necessary data from TIGER and Summary File 1.
"""

import os
import site
import sys

# Basically, add '../src' to our path.
# Needed for the imports that follow.
site.addsitedir(os.path.dirname(os.path.abspath(sys.argv[0])) + '/../src')

import Data
from Data import State, County


# Create the State objects.
de = State(10, 'Delaware', 'DE')
dc = State(11, 'District of Columbia', 'DC')
md = State(24, 'Maryland', 'MD')
nj = State(34, 'New Jersey', 'NJ')
ny = State(36, 'New York', 'NY')
pa = State(42, 'Pennsylvania', 'PA')
va = State(51, 'Virginia', 'VA')
wv = State(54, 'West Virginia', 'WV')


# Now we add all of the counties belonging to each state, starting
# with Delaware.

de.add_county(1, 'Kent')
de.add_county(3, 'New Castle')
de.add_county(5, 'Sussex')


# Washington D.C.

dc.add_county(1, 'District of Columbia', True)


# Maryland

md.add_county(1, 'Allegany')
md.add_county(3, 'Anne Arundel')
md.add_county(5, 'Baltimore')
md.add_county(510, 'Baltimore city', True)
md.add_county(9, 'Calvert')
md.add_county(11, 'Caroline')
md.add_county(13, 'Carroll')
md.add_county(15, 'Cecil')
md.add_county(17, 'Charles')
md.add_county(19, 'Dorchester')
md.add_county(21, 'Frederick')
md.add_county(23, 'Garrett')
md.add_county(25, 'Harford')
md.add_county(27, 'Howard')
md.add_county(29, 'Kent')
md.add_county(31, 'Montgomery')
md.add_county(33, 'Prince Georges')
md.add_county(35, 'Queen Annes')
md.add_county(37, 'St Marys')
md.add_county(39, 'Somerset')
md.add_county(41, 'Talbot')
md.add_county(43, 'Washington')
md.add_county(45, 'Wicomico')
md.add_county(47, 'Worcester')


# New Jersey

nj.add_county(1, 'Atlantic')
nj.add_county(3, 'Bergen')
nj.add_county(5, 'Burlington')
nj.add_county(7, 'Camden')
nj.add_county(9, 'Cape May')
nj.add_county(11, 'Cumberland')
nj.add_county(13, 'Essex')
nj.add_county(15, 'Gloucester')
nj.add_county(17, 'Hudson')
nj.add_county(19, 'Hunterdon')
nj.add_county(21, 'Mercer')
nj.add_county(23, 'Middlesex')
nj.add_county(25, 'Monmouth')
nj.add_county(27, 'Morris')
nj.add_county(29, 'Ocean')
nj.add_county(31, 'Passaic')
nj.add_county(33, 'Salem')
nj.add_county(35, 'Somerset')
nj.add_county(37, 'Sussex')
nj.add_county(39, 'Union')
nj.add_county(41, 'Warren')


# New York

ny.add_county(1, 'Albany')
ny.add_county(3, 'Allegany')
ny.add_county(5, 'Bronx')
ny.add_county(7, 'Broome')
ny.add_county(9, 'Cattaraugus')
ny.add_county(11, 'Cayuga')
ny.add_county(13, 'Chautauqua')
ny.add_county(15, 'Chemung')
ny.add_county(17, 'Chenango')
ny.add_county(19, 'Clinton')
ny.add_county(21, 'Columbia')
ny.add_county(23, 'Cortland')
ny.add_county(25, 'Delaware')
ny.add_county(27, 'Dutchess')
ny.add_county(29, 'Erie')
ny.add_county(31, 'Essex')
ny.add_county(33, 'Franklin')
ny.add_county(35, 'Fulton')
ny.add_county(37, 'Genesee')
ny.add_county(39, 'Greene')
ny.add_county(41, 'Hamilton')
ny.add_county(43, 'Herkimer')
ny.add_county(45, 'Jefferson')
ny.add_county(47, 'Kings')
ny.add_county(49, 'Lewis')
ny.add_county(51, 'Livingston')
ny.add_county(53, 'Madison')
ny.add_county(55, 'Monroe')
ny.add_county(57, 'Montgomery')
ny.add_county(59, 'Nassau')
ny.add_county(61, 'New York')
ny.add_county(63, 'Niagara')
ny.add_county(65, 'Oneida')
ny.add_county(67, 'Onondaga')
ny.add_county(69, 'Ontario')
ny.add_county(71, 'Orange')
ny.add_county(73, 'Orleans')
ny.add_county(75, 'Oswego')
ny.add_county(77, 'Otsego')
ny.add_county(79, 'Putnam')
ny.add_county(81, 'Queens')
ny.add_county(83, 'Rensselaer')
ny.add_county(85, 'Richmond')
ny.add_county(87, 'Rockland')
ny.add_county(89, 'St Lawrence')
ny.add_county(91, 'Saratoga')
ny.add_county(93, 'Schenectady')
ny.add_county(95, 'Schoharie')
ny.add_county(97, 'Schuyler')
ny.add_county(99, 'Seneca')
ny.add_county(101, 'Steuben')
ny.add_county(103, 'Suffolk')
ny.add_county(105, 'Sullivan')
ny.add_county(107, 'Tioga')
ny.add_county(109, 'Tompkins')
ny.add_county(111, 'Ulster')
ny.add_county(113, 'Warren')
ny.add_county(115, 'Washington')
ny.add_county(117, 'Wayne')
ny.add_county(119, 'Westchester')
ny.add_county(121, 'Wyoming')
ny.add_county(123, 'Yates')


# Pennsylvania

pa.add_county(1, 'Adams')
pa.add_county(3, 'Allegheny')
pa.add_county(5, 'Armstrong')
pa.add_county(7, 'Beaver')
pa.add_county(9, 'Bedford')
pa.add_county(11, 'Berks')
pa.add_county(13, 'Blair')
pa.add_county(15, 'Bradford')
pa.add_county(17, 'Bucks')
pa.add_county(19, 'Butler')
pa.add_county(21, 'Cambria')
pa.add_county(23, 'Cameron')
pa.add_county(25, 'Carbon')
pa.add_county(27, 'Centre')
pa.add_county(29, 'Chester')
pa.add_county(31, 'Clarion')
pa.add_county(33, 'Clearfield')
pa.add_county(35, 'Clinton')
pa.add_county(37, 'Columbia')
pa.add_county(39, 'Crawford')
pa.add_county(41, 'Cumberland')
pa.add_county(43, 'Dauphin')
pa.add_county(45, 'Delaware')
pa.add_county(47, 'Elk')
pa.add_county(49, 'Erie')
pa.add_county(51, 'Fayette')
pa.add_county(53, 'Forest')
pa.add_county(55, 'Franklin')
pa.add_county(57, 'Fulton')
pa.add_county(59, 'Greene')
pa.add_county(61, 'Huntingdon')
pa.add_county(63, 'Indiana')
pa.add_county(65, 'Jefferson')
pa.add_county(67, 'Juniata')
pa.add_county(69, 'Lackawanna')
pa.add_county(71, 'Lancaster')
pa.add_county(73, 'Lawrence')
pa.add_county(75, 'Lebanon')
pa.add_county(77, 'Lehigh')
pa.add_county(79, 'Luzerne')
pa.add_county(81, 'Lycoming')
pa.add_county(83, 'McKean')
pa.add_county(85, 'Mercer')
pa.add_county(87, 'Mifflin')
pa.add_county(89, 'Monroe')
pa.add_county(91, 'Montgomery')
pa.add_county(93, 'Montour')
pa.add_county(95, 'Northampton')
pa.add_county(97, 'Northumberland')
pa.add_county(99, 'Perry')
pa.add_county(101, 'Philadelphia')
pa.add_county(103, 'Pike')
pa.add_county(105, 'Potter')
pa.add_county(107, 'Schuylkill')
pa.add_county(109, 'Snyder')
pa.add_county(111, 'Somerset')
pa.add_county(113, 'Sullivan')
pa.add_county(115, 'Susquehanna')
pa.add_county(117, 'Tioga')
pa.add_county(119, 'Union')
pa.add_county(121, 'Venango')
pa.add_county(123, 'Warren')
pa.add_county(125, 'Washington')
pa.add_county(127, 'Wayne')
pa.add_county(129, 'Westmoreland')
pa.add_county(131, 'Wyoming')
pa.add_county(133, 'York')


# Virginia

va.add_county(1, 'Accomack')
va.add_county(3, 'Albemarle')
va.add_county(5, 'Alleghany')
va.add_county(7, 'Amelia')
va.add_county(9, 'Amherst')
va.add_county(11, 'Appomattox')
va.add_county(13, 'Arlington')
va.add_county(15, 'Augusta')
va.add_county(17, 'Bath')
va.add_county(19, 'Bedford')
va.add_county(21, 'Bland')
va.add_county(23, 'Botetourt')
va.add_county(25, 'Brunswick')
va.add_county(27, 'Buchanan')
va.add_county(29, 'Buckingham')
va.add_county(31, 'Campbell')
va.add_county(33, 'Caroline')
va.add_county(35, 'Carroll')
va.add_county(36, 'Charles City')
va.add_county(37, 'Charlotte')
va.add_county(41, 'Chesterfield')
va.add_county(43, 'Clarke')
va.add_county(45, 'Craig')
va.add_county(47, 'Culpeper')
va.add_county(49, 'Cumberland')
va.add_county(51, 'Dickenson')
va.add_county(53, 'Dinwiddie')
va.add_county(57, 'Essex')
va.add_county(59, 'Fairfax')
va.add_county(61, 'Fauquier')
va.add_county(63, 'Floyd')
va.add_county(65, 'Fluvanna')
va.add_county(67, 'Franklin')
va.add_county(69, 'Frederick')
va.add_county(71, 'Giles')
va.add_county(73, 'Gloucester')
va.add_county(75, 'Goochland')
va.add_county(77, 'Grayson')
va.add_county(79, 'Greene')
va.add_county(81, 'Greensville')
va.add_county(83, 'Halifax')
va.add_county(85, 'Hanover')
va.add_county(87, 'Henrico')
va.add_county(89, 'Henry')
va.add_county(91, 'Highland')
va.add_county(93, 'Isle of Wight')
va.add_county(95, 'James City')
va.add_county(97, 'King and Queen')
va.add_county(99, 'King George')
va.add_county(101, 'King William')
va.add_county(103, 'Lancaster')
va.add_county(105, 'Lee')
va.add_county(107, 'Loudoun')
va.add_county(109, 'Louisa')
va.add_county(111, 'Lunenburg')
va.add_county(113, 'Madison')
va.add_county(115, 'Mathews')
va.add_county(117, 'Mecklenburg')
va.add_county(119, 'Middlesex')
va.add_county(121, 'Montgomery')
va.add_county(125, 'Nelson')
va.add_county(127, 'New Kent')
va.add_county(131, 'Northampton')
va.add_county(133, 'Northumberland')
va.add_county(135, 'Nottoway')
va.add_county(137, 'Orange')
va.add_county(139, 'Page')
va.add_county(141, 'Patrick')
va.add_county(143, 'Pittsylvania')
va.add_county(145, 'Powhatan')
va.add_county(147, 'Prince Edward')
va.add_county(149, 'Prince George')
va.add_county(153, 'Prince William')
va.add_county(155, 'Pulaski')
va.add_county(157, 'Rappahannock')
va.add_county(159, 'Richmond')
va.add_county(161, 'Roanoke')
va.add_county(163, 'Rockbridge')
va.add_county(165, 'Rockingham')
va.add_county(167, 'Russell')
va.add_county(169, 'Scott')
va.add_county(171, 'Shenandoah')
va.add_county(173, 'Smyth')
va.add_county(175, 'Southampton')
va.add_county(177, 'Spotsylvania')
va.add_county(179, 'Stafford')
va.add_county(181, 'Surry')
va.add_county(183, 'Sussex')
va.add_county(185, 'Tazewell')
va.add_county(187, 'Warren')
va.add_county(191, 'Washington')
va.add_county(193, 'Westmoreland')
va.add_county(195, 'Wise')
va.add_county(197, 'Wythe')
va.add_county(199, 'York')
va.add_county(510, 'Alexandria city', True)
va.add_county(515, 'Bedford city', True)
va.add_county(520, 'Bristol city', True)
va.add_county(530, 'Buena Vista city', True)
va.add_county(540, 'Charlottesville city', True)
va.add_county(550, 'Chesapeake city', True)
va.add_county(570, 'Colonial Heights city', True)
va.add_county(580, 'Covington city', True)
va.add_county(590, 'Danville city', True)
va.add_county(595, 'Emporia city', True)
va.add_county(600, 'Fairfax city', True)
va.add_county(610, 'Falls Church city', True)
va.add_county(620, 'Franklin city', True)
va.add_county(630, 'Fredericksburg city', True)
va.add_county(640, 'Galax city', True)
va.add_county(650, 'Hampton city', True)
va.add_county(660, 'Harrisonburg city', True)
va.add_county(670, 'Hopewell city', True)
va.add_county(678, 'Lexington city', True)
va.add_county(680, 'Lynchburg city', True)
va.add_county(683, 'Manassas city', True)
va.add_county(685, 'Manassas Park city', True)
va.add_county(690, 'Martinsville city', True)
va.add_county(700, 'Newport News city', True)
va.add_county(710, 'Norfolk city', True)
va.add_county(720, 'Norton city', True)
va.add_county(730, 'Petersburg city', True)
va.add_county(735, 'Poquoson city', True)
va.add_county(740, 'Portsmouth city', True)
va.add_county(750, 'Radford city', True)
va.add_county(760, 'Richmond city', True)
va.add_county(770, 'Roanoke city', True)
va.add_county(775, 'Salem city', True)
va.add_county(790, 'Staunton city', True)
va.add_county(800, 'Suffolk city', True)
va.add_county(810, 'Virginia Beach city', True)
va.add_county(820, 'Waynesboro city', True)
va.add_county(830, 'Williamsburg city', True)
va.add_county(840, 'Winchester city', True)


# West Virginia

wv.add_county(1, 'Barbour')
wv.add_county(3, 'Berkeley')
wv.add_county(5, 'Boone')
wv.add_county(7, 'Braxton')
wv.add_county(9, 'Brooke')
wv.add_county(11, 'Cabell')
wv.add_county(13, 'Calhoun')
wv.add_county(15, 'Clay')
wv.add_county(17, 'Doddridge')
wv.add_county(19, 'Fayette')
wv.add_county(21, 'Gilmer')
wv.add_county(23, 'Grant')
wv.add_county(25, 'Greenbrier')
wv.add_county(27, 'Hampshire')
wv.add_county(29, 'Hancock')
wv.add_county(31, 'Hardy')
wv.add_county(33, 'Harrison')
wv.add_county(35, 'Jackson')
wv.add_county(37, 'Jefferson')
wv.add_county(39, 'Kanawha')
wv.add_county(41, 'Lewis')
wv.add_county(43, 'Lincoln')
wv.add_county(45, 'Logan')
wv.add_county(47, 'McDowell')
wv.add_county(49, 'Marion')
wv.add_county(51, 'Marshall')
wv.add_county(53, 'Mason')
wv.add_county(55, 'Mercer')
wv.add_county(57, 'Mineral')
wv.add_county(59, 'Mingo')
wv.add_county(61, 'Monongalia')
wv.add_county(63, 'Monroe')
wv.add_county(65, 'Morgan')
wv.add_county(67, 'Nicholas')
wv.add_county(69, 'Ohio')
wv.add_county(71, 'Pendleton')
wv.add_county(73, 'Pleasants')
wv.add_county(75, 'Pocahontas')
wv.add_county(77, 'Preston')
wv.add_county(79, 'Putnam')
wv.add_county(81, 'Raleigh')
wv.add_county(83, 'Randolph')
wv.add_county(85, 'Ritchie')
wv.add_county(87, 'Roane')
wv.add_county(89, 'Summers')
wv.add_county(91, 'Taylor')
wv.add_county(93, 'Tucker')
wv.add_county(95, 'Tyler')
wv.add_county(97, 'Upshur')
wv.add_county(99, 'Wayne')
wv.add_county(101, 'Webster')
wv.add_county(103, 'Wetzel')
wv.add_county(105, 'Wirt')
wv.add_county(107, 'Wood')
wv.add_county(109, 'Wyoming')


# Now that the counties have been added, create a list of states to
# pass to the download_lines function.
states = [de, dc, md, nj, ny, pa, va, wv]
Data.download_sf1(states)
Data.download_blocks(states)
Data.download_lines(states)