Add working version of the script/module

This commit is contained in:
Hugo 2021-11-13 15:57:33 +01:00
parent 5b7d526b3d
commit 81176a7ece
4 changed files with 120 additions and 1 deletions

2
.gitignore vendored
View File

@ -138,3 +138,5 @@ dmypy.json
# Cython debug symbols
cython_debug/
# Secrets
secrets.ini

View File

@ -1,6 +1,6 @@
MIT License
Copyright (c) <year> <copyright holders>
Copyright (c) 2021 Saxodwarf
Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:

1
requirements.txt Normal file
View File

@ -0,0 +1 @@
requests

116
send_sms.py Executable file
View File

@ -0,0 +1,116 @@
#!/usr/bin/env python3
""" Script to send SMSes using the Free mobile API
"""
import sys
import logging
import argparse
import readline
import configparser
from itertools import count
import requests
class SMSApiManager:
""" Class to manage ending SMS notification using Free Mobile API.
"""
FREE_API_URL = "https://smsapi.free-mobile.fr/sendmsg"
MAX_LENGTH = 999
RETURN_CODE = {200: "Message sent",
400: "Mandatory parameter is missing",
402: "Too many SMS sent in to few time",
403: "Service disabled",
500: "Server error, please try again later"}
def __init__(self, conf_file_path):
self._pass = ""
self.user_id = ""
self.read_config(conf_file_path)
def send_sms(self, msg: str):
""" Send a SMS to a user using its passphrase and user ID
Return the response object from requests.
"""
# Make sure we do not send more than MAX_LENGTH characters
msg = msg[:self.MAX_LENGTH]
data = {"user": self.user_id,
"pass": self._pass,
"msg": msg}
res = requests.get(self.FREE_API_URL, params=data)
logging.info(self.RETURN_CODE[res.status_code])
return res
def read_config(self, conf_file_path):
""" Read and apply the configuration from a config file.
That config file must have a 'creds' section, and the following options:
- 'user_id', that contains the user identifier for the API
- 'passphrase', that contains this user's API key
"""
conf_parser = configparser.ConfigParser()
conf_parser.read(conf_file_path)
self.user_id = conf_parser["creds"]["user_id"]
self._pass = conf_parser["creds"]["passphrase"]
def parse_args():
""" Parse the script's command line argument, and return the args object.
"""
my_parser = argparse.ArgumentParser(description="Script to send a SMS to a person. "
"The message must be at most "
f"{SMSApiManager.MAX_LENGTH} characters long.")
my_parser.add_argument("-c", "--config", help="Path to the config file.",
default="secrets.ini")
my_parser.add_argument("message", nargs="?",
help="Message to send to the recipient. If no message is given, "
f"read up to {SMSApiManager.MAX_LENGTH} characters from the "
"standard input")
my_parser.add_argument("-i", "--interactive", action="store_true",
help="Send messages in interactive mode "
"instead of from the command line.")
return my_parser.parse_args()
def rlinput(prompt, prefill=''):
""" Prefill the input content with the value of `prefill`
All credits go to https://stackoverflow.com/a/2533142
"""
readline.set_startup_hook(lambda: readline.insert_text(prefill))
try:
return input(prompt)
finally:
readline.set_startup_hook()
def main():
""" Instantiate an sms_manager, and send SMS
"""
args = parse_args()
sms_manager = SMSApiManager(args.config)
while args.interactive:
# Interactive mode is enabled, loop indefinetely until user hits Ctrl+C or Ctrl+D
try:
message=[]
for i in count():
message.append(rlinput(f"What is your message (line {i+1})?: ",
prefill=args.message))
args.message=""
if not message[i]:
break
res = sms_manager.send_sms(msg='\n'.join(message))
if res.status_code != 200:
print("Error while sending the message:")
print(f"{sms_manager.RETURN_CODE[res.status_code]}")
except (EOFError, KeyboardInterrupt):
print("\nBye !")
break
else:
# Else either get message from command line or stdin.
logging.basicConfig(level=logging.INFO)
if args.message:
# Send the requested message
sms_manager.send_sms(msg=args.message)
else:
# Read the message from stdin
msg = sys.stdin.read(sms_manager.MAX_LENGTH)
res = sms_manager.send_sms(msg=msg)
if __name__ == "__main__":
main()