-
Notifications
You must be signed in to change notification settings - Fork 75
/
us.py
133 lines (106 loc) · 8.38 KB
/
us.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
# TODO: move this module to some package
from datetime import datetime
# Current apportionment to the states, or "T" for territories sending a delegate
# or resident commissioner. This dict is used to filter out the historical territories
# from lists of the current states and territories.
stateapportionment = {'AL': 7, 'AK': 1, 'AS': 'T', 'AZ': 9, 'AR': 4, 'CA': 52, 'CO': 8, 'CT': 5, 'DE': 1, 'DC': 'T', 'FL': 28, 'GA': 14, 'GU': 'T', 'HI': 2, 'ID': 2, 'IL': 17, 'IN': 9, 'IA': 4, 'KS': 4, 'KY': 6, 'LA': 6, 'ME': 2, 'MD': 8, 'MA': 9, 'MI': 13, 'MN': 8, 'MS': 4, 'MO': 8, 'MT': 2, 'NE': 3, 'NV': 4, 'NH': 2, 'NJ': 12, 'NM': 3, 'NY': 26, 'NC': 14, 'ND': 1, 'MP': 'T', 'OH': 15, 'OK': 5, 'OR': 6, 'PA': 17, 'PR': 'T', 'RI': 2, 'SC': 7, 'SD': 1, 'TN': 9, 'TX': 38, 'UT': 4, 'VT': 1, 'VI': 'T', 'VA': 11, 'WA': 10, 'WV': 2, 'WI': 8, 'WY': 1}
#stateapportionment_117 = {'AL': 7, 'AK': 1, 'AS': 'T', 'AZ': 9, 'AR': 4, 'CA': 53, 'CO': 7, 'CT': 5, 'DE': 1, 'DC': 'T', 'FL': 27, 'GA': 14, 'GU': 'T', 'HI': 2, 'ID': 2, 'IL': 18, 'IN': 9, 'IA': 4, 'KS': 4, 'KY': 6, 'LA': 6, 'ME': 2, 'MD': 8, 'MA': 9, 'MI': 14, 'MN': 8, 'MS': 4, 'MO': 8, 'MT': 1, 'NE': 3, 'NV': 4, 'NH': 2, 'NJ': 12, 'NM': 3, 'NY': 27, 'NC': 13, 'ND': 1, 'MP': 'T', 'OH': 16, 'OK': 5, 'OR': 5, 'PA': 18, 'PR': 'T', 'RI': 2, 'SC': 7, 'SD': 1, 'TN': 9, 'TX': 36, 'UT': 4, 'VT': 1, 'VI': 'T', 'VA': 11, 'WA': 10, 'WV': 3, 'WI': 8, 'WY': 1}
#stateapportionment_112 = {'AL': 7, 'AK': 1, 'AS': 'T', 'AZ': 8, 'AR': 4, 'CA': 53, 'CO': 7, 'CT': 5, 'DE': 1, 'DC': 'T', 'FL': 25, 'GA': 13, 'GU': 'T', 'HI': 2, 'ID': 2, 'IL': 19, 'IN': 9, 'IA': 5, 'KS': 4, 'KY': 6, 'LA': 7, 'ME': 2, 'MD': 8, 'MA': 10, 'MI': 15, 'MN': 8, 'MS': 4, 'MO': 9, 'MT': 1, 'NE': 3, 'NV': 3, 'NH': 2, 'NJ': 13, 'NM': 3, 'NY': 29, 'NC': 13, 'ND': 1, 'MP': 'T', 'OH': 18, 'OK': 5, 'OR': 5, 'PA': 19, 'PR': 'T', 'RI': 2, 'SC': 6, 'SD': 1, 'TN': 9, 'TX': 32, 'UT': 3, 'VT': 1, 'VI': 'T', 'VA': 11, 'WA': 9, 'WV': 3, 'WI': 8, 'WY': 1}
# All state abbreviations, including historical territories.
stateabbrs = ["AL", "AK", "AS", "AZ", "AR", "CA", "CO", "CT", "DE", "DC", "FL", "GA", "GU", "HI", "ID", "IL", "IN", "IA", "KS", "KY", "LA", "ME", "MD", "MA", "MI", "MN", "MS", "MO", "MT", "NE", "NV", "NH", "NJ", "NM", "NY", "NC", "ND", "MP", "OH", "OK", "OR", "PA", "PR", "RI", "SC", "SD", "TN", "TX", "UT", "VT", "VI", "VA", "WA", "WV", "WI", "WY", "DK", "PI", "OL"]
# All state names, including historical territories.
statenames = {"AL":"Alabama", "AK":"Alaska", "AS":"American Samoa", "AZ":"Arizona", "AR":"Arkansas", "CA":"California", "CO":"Colorado", "CT":"Connecticut", "DE":"Delaware", "DC":"District of Columbia", "FL":"Florida", "GA":"Georgia", "GU":"Guam", "HI":"Hawaii", "ID":"Idaho", "IL":"Illinois", "IN":"Indiana", "IA":"Iowa", "KS":"Kansas", "KY":"Kentucky", "LA":"Louisiana", "ME":"Maine", "MD":"Maryland", "MA":"Massachusetts", "MI":"Michigan", "MN":"Minnesota", "MS":"Mississippi", "MO":"Missouri", "MT":"Montana", "NE":"Nebraska", "NV":"Nevada", "NH":"New Hampshire", "NJ":"New Jersey", "NM":"New Mexico", "NY":"New York", "NC":"North Carolina", "ND": "North Dakota", "MP":"Northern Mariana Islands", "OH":"Ohio", "OK":"Oklahoma", "OR":"Oregon", "PA":"Pennsylvania", "PR":"Puerto Rico", "RI":"Rhode Island", "SC":"South Carolina", "SD":"South Dakota", "TN":"Tennessee", "TX":"Texas", "UT":"Utah", "VT":"Vermont", "VI":"Virgin Islands", "VA":"Virginia", "WA":"Washington", "WV":"West Virginia", "WI":"Wisconsin", "WY":"Wyoming", "DK": "Dakota Territory", "PI": "Philippines", "OL": "Territory of Orleans"}
state_abbr_from_name = dict((v.lower(),k) for (k,v) in statenames.items())
# Current states, a list of (abbr, name) tuples in sorted order.
statelist = [s for s in statenames.items() if s[0] in stateapportionment]
statelist.sort(key=lambda x : x[1])
statenames_apstyle = { "AL": "Ala.", "AK": "Alaska", "AZ": "Ariz.", "AR": "Ark.", "CA": "Calif.", "CO": "Colo.", "CT": "Conn.", "DE": "Del.", "FL": "Fla.", "GA": "Ga.", "HI": "Hawaii", "ID": "Idaho", "IL": "Ill.", "IN": "Ind.", "IA": "Iowa", "KS": "Kan.", "KY": "Ky.", "LA": "La.", "ME": "Maine", "MD": "Md.", "MA": "Mass.", "MI": "Mich.", "MN": "Minn.", "MS": "Miss.", "MO": "Mo.", "MT": "Mont.", "NE": "Neb.", "NV": "Nev.", "NH": "N.H.", "NJ": "N.J.", "NM": "N.M.", "NY": "N.Y.", "NC": "N.C.", "ND": "N.D.", "OH": "Ohio", "OK": "Okla.", "OR": "Ore.", "PA": "Pa.", "RI": "R.I.", "SC": "S.C.", "SD": "S.D.", "TN": "Tenn.", "TX": "Texas", "UT": "Utah", "VT": "Vt.", "VA": "Va.", "WA": "Wash.", "WV": "W. Va.", "WI": "Wis.", "WY": "Wyo.", "DC": "District of Columbia" }
CONGRESS_DATES = {}
SESSION_DATES = []
def parse_govtrack_date(d, as_date=False):
if as_date:
return datetime.strptime(d, '%Y-%m-%d').date()
try:
return datetime.strptime(d, '%Y-%m-%dT%H:%M:%S-04:00')
except ValueError:
pass
try:
return datetime.strptime(d, '%Y-%m-%dT%H:%M:%S-05:00')
except ValueError:
pass
return datetime.strptime(d, '%Y-%m-%d')
def get_congress_dates(congressnumber):
global CONGRESS_DATES
if CONGRESS_DATES == { }:
cd = {}
for line in open('sessions.tsv'):
cn, sessionname, startdate, enddate = line.strip().split('\t')[0:4]
if not '-' in startdate: # header
continue
cn = int(cn)
if not cn in cd:
cd[cn] = [parse_govtrack_date(startdate, as_date=True), None]
cd[cn][1] = parse_govtrack_date(enddate, as_date=True)
CONGRESS_DATES.update(cd)
return CONGRESS_DATES[congressnumber]
def get_congress_years(congressnumber):
dates = get_congress_dates(congressnumber)
years = dates[0].year, dates[1].year
if dates[1].month == 1: years = (years[0], years[1]-1)
return years
def get_session_from_date(when, allow_start_date=True, allow_end_date=True, congress=None):
global SESSION_DATES
if SESSION_DATES == [ ]:
sd = []
for line in open('sessions.tsv'):
cn, sessionname, startdate, enddate = line.strip().split('\t')[0:4]
if not '-' in startdate: # header
continue
sd.append((int(cn), sessionname, parse_govtrack_date(startdate, as_date=True), parse_govtrack_date(enddate, as_date=True)))
SESSION_DATES = sd
if when == None:
return None
if isinstance(when, datetime):
when = when.date()
for c, s, sd, ed in SESSION_DATES:
if congress != None and c != congress: continue
if (sd < when or (allow_start_date and sd == when)) and (when < ed or (allow_end_date and when == ed)):
return (c, s)
return None
def get_session_ordinal(congress, session):
get_session_from_date(None) # load data
ordinal = 0
for c, s, sd, ed in SESSION_DATES:
if c == congress:
ordinal += 1
if s == session: return ordinal
raise ValueError("Congress or session not found.")
def get_all_sessions():
get_session_from_date(None) # load data
return SESSION_DATES # [(congress, session, startdate, enddate), ...]
def get_congress_from_date(when, range_type=None):
if when.isoformat() < "1941-01-03":
# Before this time the dates of Congresses were pretty arbitrary.
get_congress_dates(1) # load data
for c, (sd, ed) in CONGRESS_DATES.items():
if (sd < when or (range_type=='start' and sd == when)) and (when < ed or (range_type=='end' and when == ed)):
return c
return None
else:
# Since 1941, we can reliably use the Jan 3-Jan 3 rule.
legislative_year = when.year
if when.month == 1 and when.day <= 2:
# Get the "legislative year", which is the same as the calendar year
# except on Jan 1, 2, and half of 3 which is a part of the previous legislative year.
legislative_year = when.year - 1
elif when.month == 1 and when.day == 3:
# This date is the end of a range, so we'll treat Jan 3 as being before noon.
if range_type == "end":
legislative_year = when.year - 1
# This date is the start of a range, so we'll treat Jan 3 as being after noon.
elif range_type == "start":
legislative_year = when.year
else:
raise ValueError("Date is ambiguous; must pass range_type='start' or 'end'.")
# Now do some simple integer math to compute the Congress number.
return ((legislative_year + 1) // 2) - 894