Configuration in config file
This commit is contained in:
parent
a7da455862
commit
0816a66da3
10
bot.py
10
bot.py
|
@ -51,27 +51,31 @@ class BotListener(StreamListener):
|
||||||
sender = notification['account']['acct'] # Get sender name
|
sender = notification['account']['acct'] # Get sender name
|
||||||
sender_filename = "limiter/" + sender; # Forge file for limiter
|
sender_filename = "limiter/" + sender; # Forge file for limiter
|
||||||
if os.path.isfile(sender_filename): # Check if file exist
|
if os.path.isfile(sender_filename): # Check if file exist
|
||||||
|
log.debug("Sender file exist")
|
||||||
statbuf = os.stat(sender_filename)
|
statbuf = os.stat(sender_filename)
|
||||||
last_edit = int(statbuf.st_mtime)
|
last_edit = int(statbuf.st_mtime)
|
||||||
ts = int(time.time())
|
ts = int(time.time())
|
||||||
pprint(last_edit)
|
|
||||||
pprint(ts)
|
|
||||||
if ts - last_edit > 59: # check if file is modified 1 minute after last edition
|
if ts - last_edit > 59: # check if file is modified 1 minute after last edition
|
||||||
|
log.debug("file is too old")
|
||||||
f = open(sender_filename,'w')
|
f = open(sender_filename,'w')
|
||||||
f.write(str(1)) # reset counter
|
f.write(str(1)) # reset counter
|
||||||
f.close()
|
f.close()
|
||||||
can_continue = True
|
can_continue = True
|
||||||
else:
|
else:
|
||||||
|
log.debug("file is young")
|
||||||
f = open(sender_filename,'r+')
|
f = open(sender_filename,'r+')
|
||||||
number_of_mention = int(f.read())
|
number_of_mention = int(f.read())
|
||||||
if number_of_mention < 4: # limit of mention per minute is 4
|
if number_of_mention < 4: # limit of mention per minute is 4
|
||||||
|
log.debug("Sender have less of 4 requests")
|
||||||
f.seek(0)
|
f.seek(0)
|
||||||
f.write(str(number_of_mention + 1))
|
f.write(str(number_of_mention + 1))
|
||||||
can_continue = True
|
can_continue = True
|
||||||
else:
|
else:
|
||||||
|
log.debug("Sender have more of 4 requests")
|
||||||
can_continue = False # if number of mention is for, user can't receive anything
|
can_continue = False # if number of mention is for, user can't receive anything
|
||||||
f.close()
|
f.close()
|
||||||
else: # File not exist, create it and initialise it
|
else: # File not exist, create it and initialise it
|
||||||
|
log.debug("Sender file not exist")
|
||||||
f = open(sender_filename,"w+")
|
f = open(sender_filename,"w+")
|
||||||
f.write(str(1))
|
f.write(str(1))
|
||||||
f.close()
|
f.close()
|
||||||
|
@ -88,7 +92,7 @@ class BotListener(StreamListener):
|
||||||
if mention["acct"] != get_parameter("name_bot", config_file):
|
if mention["acct"] != get_parameter("name_bot", config_file):
|
||||||
text = text + "@" + mention["acct"] + " "
|
text = text + "@" + mention["acct"] + " "
|
||||||
|
|
||||||
media_dict = post_img(mastodon, get_parameter("default_tag", config_file), log, config_file)
|
media_dict = post_img(mastodon, get_parameter("default_text", config_file), log, config_file)
|
||||||
mastodon.status_post(text, in_reply_to_id=id, media_ids=[media_dict], sensitive=get_parameter("sensitive", config_file), visibility=visibility, spoiler_text=get_parameter("spoiler_text", config_file))
|
mastodon.status_post(text, in_reply_to_id=id, media_ids=[media_dict], sensitive=get_parameter("sensitive", config_file), visibility=visibility, spoiler_text=get_parameter("spoiler_text", config_file))
|
||||||
else:
|
else:
|
||||||
log.debug("no picture send :(")
|
log.debug("no picture send :(")
|
||||||
|
|
|
@ -1,7 +1,5 @@
|
||||||
img_path: uri_path
|
img_path: uri_path
|
||||||
name_bot: "bot"
|
name_bot: bot
|
||||||
sensitive: False
|
sensitive: False
|
||||||
default_text: "some text here"
|
default_text: some text here
|
||||||
spoiler_text: "some text here"
|
spoiler_text: some text here
|
||||||
default_tag: "tag"
|
|
||||||
|
|
||||||
|
|
Reference in a new issue