mirror of
https://github.com/Pardus-LiderAhenk/ahenk
synced 2024-11-25 21:42:21 +03:00
file transfer manager added
This commit is contained in:
parent
6b9e02104b
commit
8b9c05727f
4 changed files with 0 additions and 74 deletions
0
opt/ahenk/base/file/file_transfer_manager.py
Normal file
0
opt/ahenk/base/file/file_transfer_manager.py
Normal file
0
opt/ahenk/base/file/http_file_transfer.py
Normal file
0
opt/ahenk/base/file/http_file_transfer.py
Normal file
|
@ -1,74 +0,0 @@
|
||||||
#!/usr/bin/env python3
|
|
||||||
# -*- coding: utf-8 -*-
|
|
||||||
# Author: Volkan Şahin <volkansah.in> <bm.volkansahin@gmail.com>
|
|
||||||
|
|
||||||
import asyncio
|
|
||||||
import time
|
|
||||||
|
|
||||||
import slixmpp
|
|
||||||
|
|
||||||
from base.Scope import Scope
|
|
||||||
|
|
||||||
|
|
||||||
class FileTransfer(slixmpp.ClientXMPP):
|
|
||||||
def __init__(self, file_path):
|
|
||||||
|
|
||||||
loop = asyncio.new_event_loop()
|
|
||||||
asyncio.set_event_loop(loop)
|
|
||||||
|
|
||||||
scope = Scope().getInstance()
|
|
||||||
|
|
||||||
self.logger = scope.getLogger()
|
|
||||||
self.configuration_manager = scope.getConfigurationManager()
|
|
||||||
|
|
||||||
self.file = open(file_path, 'rb')
|
|
||||||
self.my_jid = str(self.configuration_manager.get('CONNECTION', 'uid') + '@' + self.configuration_manager.get('CONNECTION', 'servicename') + '/sender')
|
|
||||||
self.my_pass = str(self.configuration_manager.get('CONNECTION', 'password'))
|
|
||||||
self.receiver = self.configuration_manager.get('CONNECTION', 'receiverjid') + '@' + self.configuration_manager.get('CONNECTION', 'servicename') + '/Smack'
|
|
||||||
self.logger.debug('[FileTransfer] File transfer client was configured.')
|
|
||||||
|
|
||||||
slixmpp.ClientXMPP.__init__(self, self.my_jid, self.my_pass)
|
|
||||||
self.logger.debug('[FileTransfer] XMPP Client was created for file transfer.')
|
|
||||||
|
|
||||||
self.register_plugin('xep_0030') # Service Discovery
|
|
||||||
self.register_plugin('xep_0065')
|
|
||||||
|
|
||||||
self.add_event_handler("session_start", self.start)
|
|
||||||
|
|
||||||
@asyncio.coroutine
|
|
||||||
def start(self, event):
|
|
||||||
self.logger.debug('[FileTransfer] Sending file ...')
|
|
||||||
|
|
||||||
try:
|
|
||||||
# Open the S5B stream in which to write to.
|
|
||||||
proxy = yield from self['xep_0065'].handshake(self.receiver)
|
|
||||||
|
|
||||||
# Send the entire file.
|
|
||||||
i = 0
|
|
||||||
while True:
|
|
||||||
data = self.file.read(1000)
|
|
||||||
i += 1
|
|
||||||
print('-->' + str(i) + '--' + str(len(data)))
|
|
||||||
if not data:
|
|
||||||
break
|
|
||||||
yield from proxy.write(data)
|
|
||||||
|
|
||||||
time.sleep(10)
|
|
||||||
# And finally close the stream.
|
|
||||||
proxy.transport.write_eof()
|
|
||||||
|
|
||||||
except Exception as e:
|
|
||||||
self.logger.debug('[FileTransfer] A problem occurred while file transferring. Error Message: {}'.format(str(e)))
|
|
||||||
else:
|
|
||||||
self.logger.debug('[FileTransfer] File transfer is finished.')
|
|
||||||
finally:
|
|
||||||
self.logger.debug('[FileTransfer] Disconnecting file transfer resource.')
|
|
||||||
self.disconnect()
|
|
||||||
self.file.close()
|
|
||||||
self.logger.debug('[FileTransfer] Disconnected file transfer resource.')
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(file_path):
|
|
||||||
xmpp = FileTransfer(file_path)
|
|
||||||
xmpp.connect()
|
|
||||||
xmpp.process(forever=False)
|
|
Loading…
Reference in a new issue