Working prototype

test.py

@@ -0,0 +1,88 @@
+#!/usr/bin/python3
+import shelve
+import os
+from subprocess import Popen, PIPE
+
+try:
+    from StringIO import StringIO
+except ImportError:
+    from io import StringIO
+
+import subprocess
+import tempfile
+
+class Buffer:  # {{{1
+
+    """
+    This class wraps cStringIO.StringIO with two additions: The __len__
+    method and a dirty flag to determine whether a buffer has changed.
+    """
+
+    def __init__(self):
+        self.buf = StringIO()
+        self.dirty = False
+
+    def __getattr__(self, attr, default=None):
+        """ Delegate to the StringIO object. """
+        return getattr(self.buf, attr, default)
+
+    def __len__(self):
+        """ Get the total size of the buffer in bytes. """
+        position = self.buf.tell()
+        self.buf.seek(0, os.SEEK_END)
+        length = self.buf.tell()
+        self.buf.seek(position, os.SEEK_SET)
+        return length
+
+    def truncate(self, *args):
+        """ Truncate the file at the current position and set the dirty flag. """
+        if len(self) > self.buf.tell():
+            self.dirty = True
+        return self.buf.truncate(*args)
+
+    def write(self, *args):
+        """ Write a string to the file and set the dirty flag. """
+        self.dirty = True
+        return self.buf.write(*args)
+
+
+chat_id = 709766994
+user_id = 709766994
+
+file_id = '012345678910abcdef'
+
+
+def get_block_from_telegram(chat_id, digest):
+    # path = download_block(chat_id=chat_id, uid=digest)
+    # buf = open('tempfile2', 'wb')
+    buf = tempfile.TemporaryFile()
+    process = Popen(["python3.6", "download_service.py", "download", str(chat_id), str(digest)], stdout=buf, bufsize=-1,shell=False)
+    process.wait()
+    # buf.close()
+    # buf = open('tempfile2', 'rb')
+    buf.seek(0)
+    block = buf.read()
+    buf.close()
+    return block
+
+block = get_block_from_telegram(chat_id, file_id)
+outfile = open('tempfile.mp3','wb')
+outfile.write(block)
+outfile.close()
+buf = Buffer()
+buf = b'12345678'
+# with open('testfile.mp3','rb') as fp:
+#     buf = fp.read()
+#
+process = Popen(["python3.6", "download_service.py", "upload", str(chat_id), str(file_id)], stdin=PIPE, bufsize=-1)
+process.stdin.write(buf)
+process.stdin.close()
+process.wait()
+
+
+# upload_block(string_to_upload=buf.buf.getvalue(), chat_id=chat_id, hash_uid=str(file_id))
+# storage = shelve.open('./storage.db')
+# object = get_block_from_telegram(chat_id, file_id)
+#
+# storage[file_id] = object
+# print(storage[file_id])