tweaks I made while figureing out how this generates posts

This commit is contained in:
Amber 2022-03-22 15:56:35 -05:00
parent 12fb5a558d
commit adfaaf8a24
3 changed files with 8 additions and 4 deletions

View File

@ -6,5 +6,10 @@
"mention_handling": 1,
"max_thread_length": 15,
"strip_paired_punctuation": false,
"limit_length": false,
"length_lower_limit": 5,
"length_upper_limit": 50,
"overlap_ratio_enabled": false,
"overlap_ratio": 0.7,
"website": "https://git.nixnet.services/amber/amber-ebooks"
}

View File

@ -64,7 +64,7 @@ def make_toot(cfg):
pin, pout = multiprocessing.Pipe(False)
p = multiprocessing.Process(target=make_sentence, args=[pout, cfg])
p.start()
p.join(5) # wait 5 seconds to get something
p.join(300) # wait 5 seconds to get something
if p.is_alive(): # if it's still trying to make a toot after 5 seconds
p.terminate()
p.join()
@ -77,6 +77,7 @@ def make_toot(cfg):
def extract_toot(toot):
toot = re.sub("<br>", "\n", toot)
toot = html.unescape(toot) # convert HTML escape codes to text
soup = BeautifulSoup(toot, "html.parser")
for lb in soup.select("br"): # replace <br> with linebreak

View File

@ -10,9 +10,7 @@ import requests
import functions
parser = argparse.ArgumentParser(description='Log in and download posts.')
parser.add_argument(
'-c', '--cfg', dest='cfg', default='config.json', nargs='?',
help="Specify a custom location for config.json.")
parser.add_argument('-c', '--cfg', dest='cfg', default='config.json', nargs='?', help="Specify a custom location for config.json.")
args = parser.parse_args()