a/ebooks
1
1
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Remove @ mentions

Browse source
This commit is contained in:
Anthony Wang 2022-02-23 15:31:02 -06:00
parent cc60315e21
commit 4a9671c958
Signed by: a
GPG key ID: BC96B00AEC5F2D76
1 changed files with 7 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
bot.py
View file

@ -54,10 +54,17 @@ if args.input is None:
])
else:
with open(args.data, 'r') as f:
# Get a line with at least two words
lines = f.readlines()
line = choice(lines).split()
while len(line) < 2:
line = choice(lines).split()
# Remove mentions
if line[0].count('@') > 1:
line[0] = '@'.join(line[0].split('@')[0:2])
if line[1].count('@') > 1:
line[1] = '@'.join(line[1].split('@')[0:2])
args.input = line[0] + ' ' + line[1]