Merge commit '24f5bc60dcf0610d65394e0026c6fe91bdb59e67' as 'utils_v2'
This commit is contained in:
@@ -0,0 +1,284 @@
|
||||
"""
|
||||
|
||||
AUTHOR:
|
||||
|
||||
Khushal P Soonderji
|
||||
|
||||
DATE:
|
||||
|
||||
Monday, 9th Sept., 2024
|
||||
|
||||
OBJECTIVE:
|
||||
|
||||
To be able to send SMSs from Nimbus's API and manage the templates and other things from one place.
|
||||
|
||||
REFERENCES:
|
||||
|
||||
01. https://github.com/innovativevijay/SmsHitApiSample
|
||||
02. https://nimbusit.net/appforms/apimanual.php
|
||||
|
||||
DOWNLOADS:
|
||||
|
||||
N/A
|
||||
|
||||
WEB-PORTAL:
|
||||
|
||||
01. http://nimbusit.net/
|
||||
|
||||
"""
|
||||
|
||||
|
||||
# *****************************************************************************************************************
|
||||
# ***** ****
|
||||
# *** IMPORT ***
|
||||
# ***** ****
|
||||
# *****************************************************************************************************************
|
||||
|
||||
|
||||
# To make sibling directories accessible for imports:
|
||||
import sys
|
||||
sys.path.append(".")
|
||||
sys.path.append("..")
|
||||
|
||||
# To make API Calls:
|
||||
import httpx
|
||||
|
||||
# For debugging:
|
||||
from icecream import IceCreamDebugger
|
||||
|
||||
|
||||
# *****************************************************************************************************************
|
||||
# ***** ****
|
||||
# *** MACROS / ONE-TIME INIT ***
|
||||
# ***** ****
|
||||
# *****************************************************************************************************************
|
||||
|
||||
|
||||
# --- Nothing yet
|
||||
|
||||
|
||||
# *****************************************************************************************************************
|
||||
# ***** ****
|
||||
# *** VARIABLES ***
|
||||
# ***** ****
|
||||
# *****************************************************************************************************************
|
||||
|
||||
|
||||
# --- Nothing Yet
|
||||
|
||||
|
||||
# *****************************************************************************************************************
|
||||
# ***** ****
|
||||
# *** CLASSES ***
|
||||
# ***** ****
|
||||
# *****************************************************************************************************************
|
||||
|
||||
|
||||
class AsyncNimbusSMS:
|
||||
|
||||
MESSAGE_TYPE_REGULAR = 0
|
||||
MESSAGE_TYPE_UNICODE = 1
|
||||
|
||||
MESSAGE_TYPE_NOT_FLASH = 0
|
||||
MESSAGE_TYPE_FLASH = 1
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
entity_id,
|
||||
sender_id,
|
||||
user_id,
|
||||
api_key,
|
||||
http_client: httpx.AsyncClient = None,
|
||||
debug = True,
|
||||
debug_prefix = "Nimbus SMS | ",
|
||||
debug_only_errors = True
|
||||
):
|
||||
|
||||
"""
|
||||
Sets up an instance of the SMS sender through Nimbus IT.
|
||||
:param entity_id: The entity id as registered with DLT.
|
||||
:param sender_id: The 6-char code like "HDFCBK", "NSESMS", "ZRODHA" that you see in your SMS inbox.
|
||||
:param user_id: The 6-digit id that Nimbus has assigned to you.
|
||||
:param api_key: The API key generated through Nimbus's portal.
|
||||
:param http_client: An instance of 'httpx' library's AsyncClient. If not given, one will be instantiated
|
||||
internally. It is recommended that, for multi-auth use cases, you provide a common HTTP client from outside.
|
||||
:param debug: Whether, or not, you would like to show debugging messages (can be changed on the fly).
|
||||
:param debug_prefix: The prefix text to show with the debug string.
|
||||
:param debug_only_errors: Whether you would like to show all debugging messages or just error messages.
|
||||
"""
|
||||
|
||||
# Create the debugging tools:
|
||||
self.__printer = IceCreamDebugger(prefix = debug_prefix, includeContext = True)
|
||||
if not debug: self.__printer.disable()
|
||||
self.__debug_only_errors = debug_only_errors
|
||||
|
||||
# Accept/create an HTTP client to work with:
|
||||
if http_client: self.__http_client = http_client
|
||||
else: self.__http_client = httpx.AsyncClient(
|
||||
limits = httpx.Limits(
|
||||
max_connections = 100, # ............ Maximum number of connections allowed in the pool.
|
||||
max_keepalive_connections = 50, # ... Maximum number of connections that can be kept alive.
|
||||
),
|
||||
timeout = httpx.Timeout(
|
||||
pool = 120.0, # .... Time to wait for a free connection from the pool.
|
||||
connect = 2.5, # ... Time to wait for establishing a connection to the server.
|
||||
write = 10.0, # .... Time to wait for sending data.
|
||||
read = 2.5 # ....... Time to wait for receiving data.
|
||||
)
|
||||
)
|
||||
|
||||
# Capture the input config:
|
||||
self.__entity_id = entity_id
|
||||
self.__sender_id = sender_id
|
||||
self.__user_id = user_id,
|
||||
self.__api_key = api_key
|
||||
|
||||
def enable_debug(self):
|
||||
self.__printer.enable()
|
||||
|
||||
def disable_debug(self):
|
||||
self.__printer.disable()
|
||||
|
||||
def debug_only_errors(self):
|
||||
self.__debug_only_errors = True
|
||||
|
||||
def debug_everything(self):
|
||||
self.__debug_only_errors = False
|
||||
|
||||
async def get_balance(self):
|
||||
|
||||
"""
|
||||
Checks the balance in the Nimbus wallet.
|
||||
:return: The balance (float) if the request was successful, else None.
|
||||
"""
|
||||
|
||||
balance = None
|
||||
|
||||
try:
|
||||
|
||||
# Call the API:
|
||||
response = await self.__http_client.get(
|
||||
url = r"http://nimbusit.net/api/balance",
|
||||
params = {"user": self.__user_id, "authkey": self.__api_key}
|
||||
)
|
||||
|
||||
# The response of a successful API call looks like "BALANCE:599". We need just the number:
|
||||
if response.status_code in [200]: balance = float(response.content.decode().split(":")[-1].strip())
|
||||
|
||||
except Exception as exception:
|
||||
self.__printer(exception)
|
||||
|
||||
return balance
|
||||
|
||||
async def send_sms(
|
||||
self,
|
||||
template_id,
|
||||
recipient_number,
|
||||
message,
|
||||
message_type = MESSAGE_TYPE_REGULAR,
|
||||
flash = MESSAGE_TYPE_NOT_FLASH
|
||||
):
|
||||
|
||||
"""
|
||||
Sends one SMS through Nimbus IT's system. The text of the message must match the template that had been
|
||||
submitted. A mismatch may cause the message to fail at best, and raise troubles in the real-world with
|
||||
government bodies at worst. Be careful.
|
||||
:param template_id: The id of the SMS template as registered on Nimbus's portal.
|
||||
:param recipient_number: The phone number of the recipient. You can send an array of numbers, too, BUT IT IS
|
||||
STRONGLY RECOMMENDED TO NOT DO THAT TO AVOID BEING BLOCKED BY DLT.
|
||||
:param message: The message to send to the recipient. Should match the template that is being sent.
|
||||
:param message_type: Choose between 'AsyncNimbusSMS.MESSAGE_TYPE_REGULAR' (default) and
|
||||
'AsyncNimbusSMS.MESSAGE_TYPE_UNICODE' based on the type of characters being sent. Both are class variables.
|
||||
:param flash: to choose between a regular inbox SMS, or a flash SMS.
|
||||
:return: The dict of all the details of the message that was sent including whether, or not, it was successfully
|
||||
sent. Other details depend on the service provider (Nimbus IT in this case).
|
||||
"""
|
||||
|
||||
# Construct the basic structure of the response of this method:
|
||||
summary = {
|
||||
"success": False,
|
||||
"info": None,
|
||||
"sender": self.__sender_id,
|
||||
"recipient": recipient_number,
|
||||
"message": message,
|
||||
"length": len(message),
|
||||
"template_id": template_id,
|
||||
"raw": None,
|
||||
"isFlash": True if flash else False
|
||||
}
|
||||
|
||||
try:
|
||||
|
||||
# Pre-process the recipient's number:
|
||||
if not isinstance(recipient_number, (list, set, tuple)): recipient_number = [recipient_number]
|
||||
|
||||
# Call the API:
|
||||
response = await self.__http_client.get(
|
||||
url = r"http://nimbusit.net/api/pushsms",
|
||||
params = {
|
||||
"user": self.__user_id,
|
||||
"authkey": self.__api_key,
|
||||
"sender": self.__sender_id,
|
||||
"mobile": ",".join([str(num) for num in recipient_number]),
|
||||
"text": message,
|
||||
"entityid": self.__entity_id,
|
||||
"templateid": template_id,
|
||||
"type": message_type,
|
||||
"flash": flash
|
||||
}
|
||||
)
|
||||
|
||||
# For a successful API call:
|
||||
if response.status_code == 200:
|
||||
response_json = response.json()
|
||||
summary["success"] = True if response_json.get("STATUS", "ERROR").lower() in ["ok"] else False
|
||||
summary["info"] = response_json.get("RESPONSE", {}).get("INFO")
|
||||
summary["raw"] = {
|
||||
"http_code": response.status_code,
|
||||
"response": response_json,
|
||||
}
|
||||
|
||||
# For any other code that indicates some form of failure:
|
||||
else: summary["raw"] = {
|
||||
"http_code": response.status_code,
|
||||
"response": response.content.decode()
|
||||
}
|
||||
|
||||
except Exception as exception:
|
||||
self.__printer(exception)
|
||||
|
||||
return summary
|
||||
|
||||
|
||||
# *****************************************************************************************************************
|
||||
# ***** ****
|
||||
# *** MAIN PROGRAM ***
|
||||
# ***** ****
|
||||
# *****************************************************************************************************************
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
import asyncio
|
||||
|
||||
async def main():
|
||||
|
||||
sender = AsyncNimbusSMS(
|
||||
entity_id = "<your_entity_id>",
|
||||
sender_id = "<your_sender_id>",
|
||||
user_id = "<your_nimbus_user_id>",
|
||||
api_key = "<your_nimbus_api_key>"
|
||||
)
|
||||
|
||||
response = await sender.send_sms(
|
||||
template_id = "<your_sms_template_id>",
|
||||
recipient_number = "<the_number_you_want_to_send_the_message_to>",
|
||||
message = "<your_sms_message>"
|
||||
)
|
||||
print("SMS API RESPONSE:", response)
|
||||
|
||||
my_balance = await sender.get_balance()
|
||||
print("REMAINING BALANCE:", my_balance)
|
||||
|
||||
|
||||
asyncio.run(main())
|
||||
Reference in New Issue
Block a user