#!/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


dc = State(11, 'District of Columbia')
md = State(24, 'Maryland')
ny = State(36, 'New York')
pa = State(42, 'Pennsylvania')
va = State(51, 'Virginia')

# 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')

# 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')

states = [dc, md, ny, pa, va]

Data.download_lines(states)