Spaces:
Sleeping
Sleeping
File size: 1,783 Bytes
ac2020e 9ee8cbe ac2020e |
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 |
"""Text processing utilities for LegisQA"""
import re
CONGRESS_GOV_TYPE_MAP = {
"hconres": "house-concurrent-resolution",
"hjres": "house-joint-resolution",
"hr": "house-bill",
"hres": "house-resolution",
"s": "senate-bill",
"sconres": "senate-concurrent-resolution",
"sjres": "senate-joint-resolution",
"sres": "senate-resolution",
}
def escape_markdown(text: str) -> str:
"""Escape markdown special characters in text"""
MD_SPECIAL_CHARS = r"\`*_{}[]()#+-.!$"
for char in MD_SPECIAL_CHARS:
text = text.replace(char, "\\" + char)
return text
def get_sponsor_url(bioguide_id: str) -> str:
"""Generate URL for a sponsor's bioguide page"""
return f"https://bioguide.congress.gov/search/bio/{bioguide_id}"
def get_congress_gov_url(congress_num: int, legis_type: str, legis_num: int) -> str:
"""Generate Congress.gov URL for a piece of legislation"""
lt = CONGRESS_GOV_TYPE_MAP[legis_type]
return f"https://www.congress.gov/bill/{int(congress_num)}th-congress/{lt}/{int(legis_num)}"
def legis_id_to_link(legis_id: str) -> str:
"""Convert a legislation ID to a Congress.gov URL"""
congress_num, legis_type, legis_num = legis_id.split("-")
return get_congress_gov_url(congress_num, legis_type, legis_num)
def legis_id_match_to_link(matchobj):
"""Convert a regex match object to a markdown link"""
mstring = matchobj.string[matchobj.start() : matchobj.end()]
url = legis_id_to_link(mstring)
link = f"[{mstring}]({url})"
return link
def replace_legis_ids_with_urls(text: str) -> str:
"""Replace legislation IDs in text with markdown links"""
pattern = "1[12][3456789]-[a-z]+-\\d{1,5}"
rtext = re.sub(pattern, legis_id_match_to_link, text)
return rtext
|