X-Git-Url: https://git.jacobcasper.com/?p=mercuryms.git;a=blobdiff_plain;f=mms-download.py;fp=mms-download.py;h=eb03ae3b81d7fd925f22b74912c1700ced2d1390;hp=86c2a3d60454734fd79fcfb9647e6c3e866ae318;hb=e0e687e1ce8146c7ef790b0f4e93083a1662c84f;hpb=de99d6c28b6cdb8c72f79df91669598ded432a6d

diff --git a/mms-download.py b/mms-download.py
index 86c2a3d..eb03ae3 100644
--- a/mms-download.py
+++ b/mms-download.py
@@ -1,5 +1,5 @@
 import os
-import requests
+import sqlite3
 from flask import Flask, request
 from twilio.twiml.messaging_response import MessagingResponse
 
@@ -8,12 +8,17 @@ app = Flask(__name__)
 @app.route("/mms", methods=['POST'])
 def sms_reply():
     """Respond to incoming multimedia messages with a confirmed download text."""
+    sender = request.form['From']
     if request.form['NumMedia'] is not None:
         urls = [request.form['MediaUrl' + str(idx)] for idx in range(int(request.form['NumMedia']))]
-    photos = {url.split("/")[-1]: requests.get(url) for url in urls}
-    for identifier, photo in photos.items():
-        with open(identifier + '.jpg', 'wb') as f:
-            f.write(photo.content)
+
+    conn = sqlite3.connect("mercuryms.sqlite")
+    cursor = conn.cursor()
+    for url in urls:
+        cursor.execute('INSERT INTO MEDIA (PHONE_NUMBER, URI) VALUES (?, ?)', (sender, url))
+    conn.commit()
+    conn.close()
+
     resp = MessagingResponse()
     resp.message("Upload received!")
     return str(resp)