Create separate wtf module/bot
This commit is contained in:
commit
79a5b2d087
3 changed files with 206 additions and 0 deletions
34
wtf_import.py
Executable file
34
wtf_import.py
Executable file
|
|
@ -0,0 +1,34 @@
|
|||
#!/bin/env python3
|
||||
|
||||
# A short script to import the wtf program's "databases" of acronyms.
|
||||
|
||||
import asyncio
|
||||
import sys
|
||||
|
||||
import wtf
|
||||
|
||||
|
||||
async def import_file(db, acronymfile):
|
||||
with open(acronymfile) as f:
|
||||
for line in f:
|
||||
s = line.split("\t", 1)
|
||||
if len(s) == 2:
|
||||
acronym, explanation = s
|
||||
explanation = explanation.strip()
|
||||
print(f"{acronym} - {explanation}")
|
||||
await db.add(acronym, explanation, "importer")
|
||||
|
||||
def main(dbfile, acronymfiles):
|
||||
db = wtf.WtfDB(dbfile)
|
||||
loop = asyncio.get_event_loop()
|
||||
|
||||
for acronymfile in acronymfiles:
|
||||
loop.run_until_complete(import_file(db, acronymfile))
|
||||
|
||||
if __name__ == "__main__":
|
||||
if len(sys.argv) >= 3:
|
||||
main(sys.argv[1], sys.argv[2:])
|
||||
else:
|
||||
print(" USAGE:")
|
||||
print(f"{sys.argv[0]} <dbfile> <acronymfile> [<acronymfile> ...]")
|
||||
exit(1)
|
||||
Loading…
Add table
Add a link
Reference in a new issue