typing
This commit is contained in:
parent
cd3b636393
commit
ec39c353d6
316
wrapper/tox.py
316
wrapper/tox.py
File diff suppressed because it is too large
Load Diff
@ -3,16 +3,17 @@
|
||||
from ctypes import (CFUNCTYPE, POINTER, ArgumentError, byref, c_bool, c_char_p,
|
||||
c_int, c_int32, c_size_t, c_uint8, c_uint16, c_uint32,
|
||||
c_void_p, cast)
|
||||
from typing import Union, Callable
|
||||
|
||||
from wrapper.libtox import LibToxAV
|
||||
from wrapper.toxav_enums import *
|
||||
|
||||
|
||||
def LOG_ERROR(a): print('EROR> '+a)
|
||||
def LOG_WARN(a): print('WARN> '+a)
|
||||
def LOG_INFO(a): print('INFO> '+a)
|
||||
def LOG_DEBUG(a): print('DBUG> '+a)
|
||||
def LOG_TRACE(a): pass # print('DEBUGx: '+a)
|
||||
def LOG_ERROR(a: str) -> None: print('EROR> '+a)
|
||||
def LOG_WARN(a: str) -> None: print('WARN> '+a)
|
||||
def LOG_INFO(a: str) -> None: print('INFO> '+a)
|
||||
def LOG_DEBUG(a: str) -> None: print('DBUG> '+a)
|
||||
def LOG_TRACE(a: str) -> None: pass # print('DEBUGx: '+a)
|
||||
|
||||
class ToxAV:
|
||||
"""
|
||||
@ -49,7 +50,7 @@ class ToxAV:
|
||||
self.video_receive_frame_cb = None
|
||||
self.call_cb = None
|
||||
|
||||
def kill(self):
|
||||
def kill(self) -> None:
|
||||
"""
|
||||
Releases all resources associated with the A/V session.
|
||||
|
||||
@ -58,7 +59,7 @@ class ToxAV:
|
||||
"""
|
||||
self.libtoxav.toxav_kill(self._toxav_pointer)
|
||||
|
||||
def get_tox_pointer(self):
|
||||
def get_tox_pointer(self) -> None:
|
||||
"""
|
||||
Returns the Tox instance the A/V object was created for.
|
||||
|
||||
@ -69,16 +70,16 @@ class ToxAV:
|
||||
|
||||
# A/V event loop
|
||||
|
||||
def iteration_interval(self):
|
||||
def iteration_interval(self) -> int:
|
||||
"""
|
||||
Returns the interval in milliseconds when the next toxav_iterate call should be. If no call is active at the
|
||||
moment, this function returns 200.
|
||||
|
||||
:return: interval in milliseconds
|
||||
"""
|
||||
return self.libtoxav.toxav_iteration_interval(self._toxav_pointer)
|
||||
return int(self.libtoxav.toxav_iteration_interval(self._toxav_pointer))
|
||||
|
||||
def iterate(self):
|
||||
def iterate(self) -> None:
|
||||
"""
|
||||
Main loop for the session. This function needs to be called in intervals of toxav_iteration_interval()
|
||||
milliseconds. It is best called in the separate thread from tox_iterate.
|
||||
@ -87,7 +88,7 @@ class ToxAV:
|
||||
|
||||
# Call setup
|
||||
|
||||
def call(self, friend_number, audio_bit_rate, video_bit_rate):
|
||||
def call(self, friend_number: int, audio_bit_rate: int, video_bit_rate: int) -> None:
|
||||
"""
|
||||
Call a friend. This will start ringing the friend.
|
||||
|
||||
@ -121,7 +122,7 @@ class ToxAV:
|
||||
elif toxav_err_call == TOXAV_ERR_CALL['INVALID_BIT_RATE']:
|
||||
raise ArgumentError('Audio or video bit rate is invalid.')
|
||||
|
||||
def callback_call(self, callback, user_data):
|
||||
def callback_call(self, callback: Callable, user_data) -> None:
|
||||
"""
|
||||
Set the callback for the `call` event. Pass None to unset.
|
||||
|
||||
@ -143,7 +144,7 @@ class ToxAV:
|
||||
self.call_cb = c_callback(callback)
|
||||
self.libtoxav.toxav_callback_call(self._toxav_pointer, self.call_cb, user_data)
|
||||
|
||||
def answer(self, friend_number, audio_bit_rate, video_bit_rate):
|
||||
def answer(self, friend_number: int, audio_bit_rate: int, video_bit_rate: int) -> None:
|
||||
"""
|
||||
Accept an incoming call.
|
||||
|
||||
@ -157,8 +158,11 @@ class ToxAV:
|
||||
"""
|
||||
toxav_err_answer = c_int()
|
||||
LOG_DEBUG(f"toxav_answer")
|
||||
result = self.libtoxav.toxav_answer(self._toxav_pointer, c_uint32(friend_number), c_uint32(audio_bit_rate),
|
||||
c_uint32(video_bit_rate), byref(toxav_err_answer))
|
||||
result = self.libtoxav.toxav_answer(self._toxav_pointer,
|
||||
c_uint32(friend_number),
|
||||
c_uint32(audio_bit_rate),
|
||||
c_uint32(video_bit_rate),
|
||||
byref(toxav_err_answer))
|
||||
toxav_err_answer = toxav_err_answer.value
|
||||
if toxav_err_answer == TOXAV_ERR_ANSWER['OK']:
|
||||
return bool(result)
|
||||
@ -177,7 +181,7 @@ class ToxAV:
|
||||
|
||||
# Call state graph
|
||||
|
||||
def callback_call_state(self, callback, user_data):
|
||||
def callback_call_state(self, callback: Callable, user_data) -> None:
|
||||
"""
|
||||
Set the callback for the `call_state` event. Pass None to unset.
|
||||
|
||||
@ -202,7 +206,7 @@ class ToxAV:
|
||||
|
||||
# Call control
|
||||
|
||||
def call_control(self, friend_number, control):
|
||||
def call_control(self, friend_number: int, control: int) -> None:
|
||||
"""
|
||||
Sends a call control command to a friend.
|
||||
|
||||
@ -232,7 +236,7 @@ class ToxAV:
|
||||
|
||||
# A/V sending
|
||||
|
||||
def audio_send_frame(self, friend_number, pcm, sample_count, channels, sampling_rate):
|
||||
def audio_send_frame(self, friend_number: int, pcm, sample_count: int, channels: int, sampling_rate: int) -> None:
|
||||
"""
|
||||
Send an audio frame to a friend.
|
||||
|
||||
@ -277,7 +281,7 @@ class ToxAV:
|
||||
elif toxav_err_send_frame == TOXAV_ERR_SEND_FRAME['RTP_FAILED']:
|
||||
RuntimeError('Failed to push frame through rtp interface.')
|
||||
|
||||
def video_send_frame(self, friend_number, width, height, y, u, v):
|
||||
def video_send_frame(self, friend_number: int, width: int, height: int, y, u, v) -> None:
|
||||
"""
|
||||
Send a video frame to a friend.
|
||||
|
||||
@ -294,8 +298,13 @@ class ToxAV:
|
||||
"""
|
||||
toxav_err_send_frame = c_int()
|
||||
LOG_TRACE(f"toxav_video_send_frame")
|
||||
result = self.libtoxav.toxav_video_send_frame(self._toxav_pointer, c_uint32(friend_number), c_uint16(width),
|
||||
c_uint16(height), c_char_p(y), c_char_p(u), c_char_p(v),
|
||||
result = self.libtoxav.toxav_video_send_frame(self._toxav_pointer,
|
||||
c_uint32(friend_number),
|
||||
c_uint16(width),
|
||||
c_uint16(height),
|
||||
c_char_p(y),
|
||||
c_char_p(u),
|
||||
c_char_p(v),
|
||||
byref(toxav_err_send_frame))
|
||||
toxav_err_send_frame = toxav_err_send_frame.value
|
||||
if toxav_err_send_frame == TOXAV_ERR_SEND_FRAME['OK']:
|
||||
@ -319,7 +328,7 @@ class ToxAV:
|
||||
|
||||
# A/V receiving
|
||||
|
||||
def callback_audio_receive_frame(self, callback, user_data):
|
||||
def callback_audio_receive_frame(self, callback: Callable, user_data) -> None:
|
||||
"""
|
||||
Set the callback for the `audio_receive_frame` event. Pass None to unset.
|
||||
|
||||
@ -346,7 +355,7 @@ class ToxAV:
|
||||
self.audio_receive_frame_cb = c_callback(callback)
|
||||
self.libtoxav.toxav_callback_audio_receive_frame(self._toxav_pointer, self.audio_receive_frame_cb, user_data)
|
||||
|
||||
def callback_video_receive_frame(self, callback, user_data):
|
||||
def callback_video_receive_frame(self, callback: Callable, user_data) -> None:
|
||||
"""
|
||||
Set the callback for the `video_receive_frame` event. Pass None to unset.
|
||||
|
||||
|
479
wrapper/toxygen_echo.py
Normal file
479
wrapper/toxygen_echo.py
Normal file
@ -0,0 +1,479 @@
|
||||
#!/var/local/bin/python3.bash
|
||||
# -*- mode: python; indent-tabs-mode: nil; py-indent-offset: 4; coding: utf-8 -*-
|
||||
|
||||
# A work in progress - chat works.
|
||||
""" echo.py features
|
||||
- accept friend request
|
||||
- echo back friend message
|
||||
# - accept and answer friend call request
|
||||
# - send back friend audio/video data
|
||||
# - send back files friend sent
|
||||
"""
|
||||
|
||||
import sys
|
||||
import os
|
||||
import traceback
|
||||
import threading
|
||||
import random
|
||||
from ctypes import *
|
||||
import argparse
|
||||
|
||||
from time import sleep
|
||||
from os.path import exists
|
||||
|
||||
# LOG=util.log
|
||||
global LOG
|
||||
import logging
|
||||
# log = lambda x: LOG.info(x)
|
||||
LOG = logging.getLogger('app')
|
||||
def LOG_error(a): print('EROR_ '+a)
|
||||
def LOG_warn(a): print('WARN_ '+a)
|
||||
def LOG_info(a): print('INFO_ '+a)
|
||||
def LOG_debug(a): print('DBUG_ '+a)
|
||||
def LOG_trace(a): pass # print('TRAC_ '+a)
|
||||
|
||||
import wrapper
|
||||
import wrapper.toxcore_enums_and_consts as enums
|
||||
from wrapper.tox import Tox, UINT32_MAX, ToxError
|
||||
from wrapper.toxcore_enums_and_consts import TOX_CONNECTION, TOX_USER_STATUS, \
|
||||
TOX_MESSAGE_TYPE, TOX_PUBLIC_KEY_SIZE, TOX_FILE_CONTROL
|
||||
|
||||
import wrapper_tests.support_testing as ts
|
||||
from wrapper_tests.support_testing import oMainArgparser
|
||||
|
||||
import time
|
||||
def sleep(fSec):
|
||||
if 'QtCore' in globals():
|
||||
if fSec > .000001: QtCore.QThread.msleep(fSec)
|
||||
QtCore.QCoreApplication.processEvents()
|
||||
else:
|
||||
time.sleep(fSec)
|
||||
|
||||
try:
|
||||
import coloredlogs
|
||||
if 'COLOREDLOGS_LEVEL_STYLES' not in os.environ:
|
||||
os.environ['COLOREDLOGS_LEVEL_STYLES'] = 'spam=22;debug=28;verbose=34;notice=220;warning=202;success=118,bold;error=124;critical=background=red'
|
||||
except ImportError as e:
|
||||
# logging.log(logging.DEBUG, f"coloredlogs not available: {e}")
|
||||
coloredlogs = None
|
||||
|
||||
import wrapper_tests.support_testing as ts
|
||||
if 'USER' in os.environ:
|
||||
sDATA_FILE = '/tmp/logging_toxygen_' +os.environ['USER'] +'.tox'
|
||||
elif 'USERNAME' in os.environ:
|
||||
sDATA_FILE = '/tmp/logging_toxygen_' +os.environ['USERNAME'] +'.tox'
|
||||
else:
|
||||
sDATA_FILE = '/tmp/logging_toxygen_' +'data' +'.tox'
|
||||
|
||||
bHAVE_AV = False
|
||||
iDHT_TRIES = 100
|
||||
iDHT_TRY = 0
|
||||
|
||||
#?SERVER = lLOCAL[-1]
|
||||
|
||||
if not bHAVE_AV:
|
||||
class AV(): pass
|
||||
else:
|
||||
class AV(wrapper.tox.ToxAV):
|
||||
def __init__(self, core):
|
||||
super(AV, self).__init__(core)
|
||||
self.core = self.get_tox()
|
||||
|
||||
def on_call(self, fid, audio_enabled, video_enabled):
|
||||
LOG.info("Incoming %s call from %d:%s ..." % (
|
||||
"video" if video_enabled else "audio", fid,
|
||||
self.core.friend_get_name(fid)))
|
||||
bret = self.answer(fid, 48, 64)
|
||||
LOG.info(f"Answered, in call... {bret!s}")
|
||||
|
||||
def on_call_state(self, fid, state):
|
||||
LOG.info('call state:fn=%d, state=%d' % (fid, state))
|
||||
|
||||
def on_audio_bit_rate(self, fid, audio_bit_rate):
|
||||
LOG.info('audio bit rate status: fn=%d, abr=%d' %
|
||||
(fid, audio_bit_rate))
|
||||
|
||||
def on_video_bit_rate(self, fid, video_bit_rate):
|
||||
LOG.info('video bit rate status: fn=%d, vbr=%d' %
|
||||
(fid, video_bit_rate))
|
||||
|
||||
def on_audio_receive_frame(self, fid, pcm, sample_count,
|
||||
channels, sampling_rate):
|
||||
# LOG.info('audio frame: %d, %d, %d, %d' %
|
||||
# (fid, sample_count, channels, sampling_rate))
|
||||
# LOG.info('pcm len:%d, %s' % (len(pcm), str(type(pcm))))
|
||||
sys.stdout.write('.')
|
||||
sys.stdout.flush()
|
||||
bret = self.audio_send_frame(fid, pcm, sample_count,
|
||||
channels, sampling_rate)
|
||||
if bret is False:
|
||||
LOG.error('on_audio_receive_frame error.')
|
||||
|
||||
def on_video_receive_frame(self, fid, width, height, frame, u, v):
|
||||
LOG.info('video frame: %d, %d, %d, ' % (fid, width, height))
|
||||
sys.stdout.write('*')
|
||||
sys.stdout.flush()
|
||||
bret = self.video_send_frame(fid, width, height, frame, u, v)
|
||||
if bret is False:
|
||||
LOG.error('on_video_receive_frame error.')
|
||||
|
||||
def witerate(self):
|
||||
self.iterate()
|
||||
|
||||
|
||||
def save_to_file(tox, fname):
|
||||
data = tox.get_savedata()
|
||||
with open(fname, 'wb') as f:
|
||||
f.write(data)
|
||||
|
||||
def load_from_file(fname):
|
||||
assert os.path.exists(fname)
|
||||
return open(fname, 'rb').read()
|
||||
|
||||
class EchoBot():
|
||||
def __init__(self, oTox):
|
||||
self._tox = oTox
|
||||
self._tox.self_set_name("PyEchoBot")
|
||||
LOG.info(f'ID: {self._tox.self_get_address()}')
|
||||
|
||||
self.files = {}
|
||||
self.av = None
|
||||
self.on_connection_status = None
|
||||
|
||||
def start(self):
|
||||
self.connect()
|
||||
if bHAVE_AV:
|
||||
# RuntimeError: Attempted to create a second session for the same Tox instance.
|
||||
|
||||
self.av = True # AV(self._tox_pointer)
|
||||
def bobs_on_friend_request(iTox,
|
||||
public_key,
|
||||
message_data,
|
||||
message_data_size,
|
||||
*largs):
|
||||
key = ''.join(chr(x) for x in public_key[:TOX_PUBLIC_KEY_SIZE])
|
||||
sPk = wrapper.tox.bin_to_string(key, TOX_PUBLIC_KEY_SIZE)
|
||||
sMd = str(message_data, 'UTF-8')
|
||||
LOG.debug('on_friend_request ' +sPk +' ' +sMd)
|
||||
self.on_friend_request(sPk, sMd)
|
||||
LOG.info('setting bobs_on_friend_request')
|
||||
self._tox.callback_friend_request(bobs_on_friend_request)
|
||||
|
||||
def bobs_on_friend_message(iTox,
|
||||
iFriendNum,
|
||||
iMessageType,
|
||||
message_data,
|
||||
message_data_size,
|
||||
*largs):
|
||||
sMd = str(message_data, 'UTF-8')
|
||||
LOG_debug(f"on_friend_message {iFriendNum}" +' ' +sMd)
|
||||
self.on_friend_message(iFriendNum, iMessageType, sMd)
|
||||
LOG.info('setting bobs_on_friend_message')
|
||||
self._tox.callback_friend_message(bobs_on_friend_message)
|
||||
|
||||
def bobs_on_file_chunk_request(iTox, fid, filenumber, position, length, *largs):
|
||||
if length == 0:
|
||||
return
|
||||
|
||||
data = self.files[(fid, filenumber)]['f'][position:(position + length)]
|
||||
self._tox.file_send_chunk(fid, filenumber, position, data)
|
||||
self._tox.callback_file_chunk_request(bobs_on_file_chunk_request)
|
||||
|
||||
def bobs_on_file_recv(iTox, fid, filenumber, kind, size, filename, *largs):
|
||||
LOG_info(f"on_file_recv {fid!s} {filenumber!s} {kind!s} {size!s} {filename}")
|
||||
if size == 0:
|
||||
return
|
||||
self.files[(fid, filenumber)] = {
|
||||
'f': bytes(),
|
||||
'filename': filename,
|
||||
'size': size
|
||||
}
|
||||
self._tox.file_control(fid, filenumber, TOX_FILE_CONTROL['RESUME'])
|
||||
|
||||
|
||||
def connect(self):
|
||||
if not self.on_connection_status:
|
||||
def on_connection_status(iTox, iCon, *largs):
|
||||
LOG_info('ON_CONNECTION_STATUS - CONNECTED ' + repr(iCon))
|
||||
self._tox.callback_self_connection_status(on_connection_status)
|
||||
LOG.info('setting on_connection_status callback ')
|
||||
self.on_connection_status = on_connection_status
|
||||
if self._oargs.network in ['newlocal', 'local']:
|
||||
LOG.info('connecting on the new network ')
|
||||
sNet = 'newlocal'
|
||||
elif self._oargs.network == 'new':
|
||||
LOG.info('connecting on the new network ')
|
||||
sNet = 'new'
|
||||
else: # main old
|
||||
LOG.info('connecting on the old network ')
|
||||
sNet = 'old'
|
||||
sFile = self._oargs.nodes_json
|
||||
lNodes = ts.generate_nodes_from_file(sFile)
|
||||
lElts = lNodes
|
||||
random.shuffle(lElts)
|
||||
for lElt in lElts[:10]:
|
||||
status = self._tox.self_get_connection_status()
|
||||
try:
|
||||
if self._tox.bootstrap(*lElt):
|
||||
LOG.info('connected to ' + lElt[0]+' '+repr(status))
|
||||
else:
|
||||
LOG.warn('failed connecting to ' + lElt[0])
|
||||
except Exception as e:
|
||||
LOG.warn('error connecting to ' + lElt[0])
|
||||
|
||||
if self._oargs.proxy_type > 0:
|
||||
random.shuffle(lElts)
|
||||
for lElt in lElts[:10]:
|
||||
status = self._tox.self_get_connection_status()
|
||||
try:
|
||||
if self._tox.add_tcp_relay(*lElt):
|
||||
LOG.info('relayed to ' + lElt[0] +' '+repr(status))
|
||||
else:
|
||||
LOG.warn('failed relay to ' + lElt[0])
|
||||
except Exception as e:
|
||||
LOG.warn('error relay to ' + lElt[0])
|
||||
|
||||
def loop(self):
|
||||
if not self.av:
|
||||
self.start()
|
||||
checked = False
|
||||
save_to_file(self._tox, sDATA_FILE)
|
||||
|
||||
LOG.info('Starting loop.')
|
||||
while True:
|
||||
|
||||
status = self._tox.self_get_connection_status()
|
||||
if not checked and status:
|
||||
LOG.info('Connected to DHT.')
|
||||
checked = True
|
||||
if not checked and not status:
|
||||
global iDHT_TRY
|
||||
iDHT_TRY += 10
|
||||
self.connect()
|
||||
self.iterate(100)
|
||||
if iDHT_TRY >= iDHT_TRIES:
|
||||
raise RuntimeError("Failed to connect to the DHT.")
|
||||
LOG.warn(f"NOT Connected to DHT. {iDHT_TRY}")
|
||||
checked = True
|
||||
if checked and not status:
|
||||
LOG.info('Disconnected from DHT.')
|
||||
self.connect()
|
||||
checked = False
|
||||
|
||||
if bHAVE_AV:
|
||||
True # self.av.witerate()
|
||||
self.iterate(100)
|
||||
|
||||
LOG.info('Ending loop.')
|
||||
|
||||
def iterate(self, n=100):
|
||||
interval = self._tox.iteration_interval()
|
||||
for i in range(n):
|
||||
self._tox.iterate()
|
||||
sleep(interval / 1000.0)
|
||||
self._tox.iterate()
|
||||
|
||||
def on_friend_request(self, pk, message):
|
||||
LOG.debug('Friend request from %s: %s' % (pk, message))
|
||||
self._tox.friend_add_norequest(pk)
|
||||
LOG.info('on_friend_request Accepted.')
|
||||
save_to_file(self._tox, sDATA_FILE)
|
||||
|
||||
def on_friend_message(self, friendId, type, message):
|
||||
name = self._tox.friend_get_name(friendId)
|
||||
LOG.debug('%s: %s' % (name, message))
|
||||
yMessage = bytes(message, 'UTF-8')
|
||||
self._tox.friend_send_message(friendId, TOX_MESSAGE_TYPE['NORMAL'], yMessage)
|
||||
LOG.info('EchoBot sent: %s' % message)
|
||||
|
||||
def on_file_recv_chunk(self, fid, filenumber, position, data):
|
||||
filename = self.files[(fid, filenumber)]['filename']
|
||||
size = self.files[(fid, filenumber)]['size']
|
||||
LOG.debug(f"on_file_recv_chunk {fid!s} {filenumber!s} {filename} {position/float(size)*100!s}")
|
||||
|
||||
if data is None:
|
||||
msg = "I got '{}', sending it back right away!".format(filename)
|
||||
self._tox.friend_send_message(fid, TOX_MESSAGE_TYPE['NORMAL'], msg)
|
||||
|
||||
self.files[(fid, 0)] = self.files[(fid, filenumber)]
|
||||
|
||||
length = self.files[(fid, filenumber)]['size']
|
||||
self.file_send(fid, 0, length, filename, filename)
|
||||
|
||||
del self.files[(fid, filenumber)]
|
||||
return
|
||||
|
||||
self.files[(fid, filenumber)]['f'] += data
|
||||
|
||||
class App():
|
||||
def __init__(self):
|
||||
self.mode = 0
|
||||
oAPP = App()
|
||||
|
||||
class EchobotTox(Tox):
|
||||
|
||||
def __init__(self, opts, app=None):
|
||||
|
||||
super(EchobotTox, self).__init__(opts, app=app)
|
||||
self._address = self.self_get_address()
|
||||
self.name = 'pyechobot'
|
||||
self._opts = opts
|
||||
self._app = app
|
||||
|
||||
class BaseThread(threading.Thread):
|
||||
|
||||
def __init__(self, name=None, target=None):
|
||||
if name:
|
||||
super().__init__(name=name, target=target)
|
||||
else:
|
||||
super().__init__(target=target)
|
||||
self._stop_thread = False
|
||||
self.name = name
|
||||
|
||||
def stop_thread(self, timeout=-1):
|
||||
self._stop_thread = True
|
||||
if timeout < 0:
|
||||
timeout = ts.iTHREAD_TIMEOUT
|
||||
i = 0
|
||||
while i < ts.iTHREAD_JOINS:
|
||||
self.join(timeout)
|
||||
if not self.is_alive(): break
|
||||
i = i + 1
|
||||
else:
|
||||
LOG.warning(f"{self.name} BLOCKED")
|
||||
|
||||
class ToxIterateThread(BaseThread):
|
||||
|
||||
def __init__(self, tox):
|
||||
super().__init__(name='ToxIterateThread')
|
||||
self._tox = tox
|
||||
|
||||
def run(self):
|
||||
while not self._stop_thread:
|
||||
self._tox.iterate()
|
||||
sleep(self._tox.iteration_interval() / 1000)
|
||||
|
||||
def oArgparse(lArgv):
|
||||
parser = ts.oMainArgparser()
|
||||
parser.add_argument('--norequest',type=str, default='False',
|
||||
choices=['True','False'],
|
||||
help='Use _norequest')
|
||||
parser.add_argument('profile', type=str, nargs='?', default=None,
|
||||
help='Path to Tox profile')
|
||||
oArgs = parser.parse_args(lArgv)
|
||||
|
||||
for key in ts.lBOOLEANS:
|
||||
if key not in oArgs: continue
|
||||
val = getattr(oArgs, key)
|
||||
setattr(oArgs, key, bool(val))
|
||||
|
||||
if hasattr(oArgs, 'sleep'):
|
||||
if oArgs.sleep == 'qt':
|
||||
pass # broken or gevent.sleep(idle_period)
|
||||
elif oArgs.sleep == 'gevent':
|
||||
pass # broken or gevent.sleep(idle_period)
|
||||
else:
|
||||
oArgs.sleep = 'time'
|
||||
|
||||
return oArgs
|
||||
|
||||
def iMain(oArgs):
|
||||
global sDATA_FILE
|
||||
# oTOX_OPTIONS = ToxOptions()
|
||||
global oTOX_OPTIONS
|
||||
oMainArgparser
|
||||
oTOX_OPTIONS = ts.oToxygenToxOptions(oArgs)
|
||||
opts = oTOX_OPTIONS
|
||||
if coloredlogs:
|
||||
coloredlogs.install(
|
||||
level=oArgs.loglevel,
|
||||
logger=LOG,
|
||||
# %(asctime)s,%(msecs)03d %(hostname)s [%(process)d]
|
||||
fmt='%(name)s %(levelname)s %(message)s'
|
||||
)
|
||||
else:
|
||||
if 'logfile' in oArgs:
|
||||
logging.basicConfig(filename=oArgs.logfile,
|
||||
level=oArgs.loglevel,
|
||||
format='%(levelname)-8s %(message)s')
|
||||
else:
|
||||
logging.basicConfig(level=oArgs.loglevel,
|
||||
format='%(levelname)-8s %(message)s')
|
||||
|
||||
iRet = 0
|
||||
if hasattr(oArgs,'profile') and oArgs.profile and os.path.isfile(oArgs.profile):
|
||||
sDATA_FILE = oArgs.profile
|
||||
LOG.info(f"loading from {sDATA_FILE}")
|
||||
opts.savedata_data = load_from_file(sDATA_FILE)
|
||||
opts.savedata_length = len(opts.savedata_data)
|
||||
opts.savedata_type = enums.TOX_SAVEDATA_TYPE['TOX_SAVE']
|
||||
else:
|
||||
opts.savedata_data = None
|
||||
|
||||
try:
|
||||
oTox = EchobotTox(opts, app=oAPP)
|
||||
t = EchoBot(oTox)
|
||||
t._oargs = oArgs
|
||||
t.start()
|
||||
t.loop()
|
||||
save_to_file(t._tox, sDATA_FILE)
|
||||
except KeyboardInterrupt:
|
||||
save_to_file(t._tox, sDATA_FILE)
|
||||
except RuntimeError as e:
|
||||
LOG.error(f"ERROR {e}")
|
||||
iRet = 1
|
||||
except Exception as e:
|
||||
LOG.error(f"EXCEPTION {e}")
|
||||
LOG.warn(' iMain(): ' \
|
||||
+'\n' + traceback.format_exc())
|
||||
iRet = 1
|
||||
return iRet
|
||||
|
||||
def main(largs=None):
|
||||
if largs is None: largs = []
|
||||
oArgs = oArgparse(largs)
|
||||
global oTOX_OARGS
|
||||
oTOX_OARGS = oArgs
|
||||
print(oArgs)
|
||||
|
||||
if coloredlogs:
|
||||
logger = logging.getLogger()
|
||||
# https://pypi.org/project/coloredlogs/
|
||||
coloredlogs.install(level=oArgs.loglevel,
|
||||
logger=logger,
|
||||
# %(asctime)s,%(msecs)03d %(hostname)s [%(process)d]
|
||||
fmt='%(name)s %(levelname)s %(message)s'
|
||||
)
|
||||
else:
|
||||
logging.basicConfig(level=oArgs.loglevel) # logging.INFO
|
||||
|
||||
return iMain(oArgs)
|
||||
|
||||
|
||||
def main(lArgs=None):
|
||||
global oTOX_OARGS
|
||||
if lArgs is None: lArgs = []
|
||||
oArgs = oArgparse(lArgs)
|
||||
global bIS_LOCAL
|
||||
bIS_LOCAL = oArgs.network in ['newlocal', 'localnew', 'local']
|
||||
oTOX_OARGS = oArgs
|
||||
setattr(oTOX_OARGS, 'bIS_LOCAL', bIS_LOCAL)
|
||||
bIS_LOCAL = True
|
||||
setattr(oTOX_OARGS, 'bIS_LOCAL', bIS_LOCAL)
|
||||
# oTOX_OPTIONS = ToxOptions()
|
||||
global oTOX_OPTIONS
|
||||
oTOX_OPTIONS = ts.oToxygenToxOptions(oArgs)
|
||||
if coloredlogs:
|
||||
# https://pypi.org/project/coloredlogs/
|
||||
coloredlogs.install(level=oArgs.loglevel,
|
||||
logger=LOG,
|
||||
# %(asctime)s,%(msecs)03d %(hostname)s [%(process)d]
|
||||
fmt='%(name)s %(levelname)s %(message)s'
|
||||
)
|
||||
else:
|
||||
logging.basicConfig(level=oArgs.loglevel) # logging.INFO
|
||||
|
||||
return iMain(oArgs)
|
||||
|
||||
if __name__ == '__main__':
|
||||
sys.exit(main(sys.argv[1:]))
|
@ -37,7 +37,7 @@ except ImportError as e:
|
||||
nmap = False
|
||||
|
||||
import wrapper
|
||||
from wrapper.toxcore_enums_and_consts import TOX_CONNECTION, TOX_USER_STATUS
|
||||
import wrapper.toxcore_enums_and_consts as enums
|
||||
|
||||
from wrapper_tests.support_http import bAreWeConnected
|
||||
from wrapper_tests.support_onions import (is_valid_fingerprint,
|
||||
@ -281,6 +281,53 @@ def get_audio():
|
||||
'enabled': input_devices and output_devices}
|
||||
return audio
|
||||
|
||||
def oToxygenToxOptions(oArgs):
|
||||
data = None
|
||||
tox_options = wrapper.tox.Tox.options_new()
|
||||
if oArgs.proxy_type:
|
||||
tox_options.contents.proxy_type = int(oArgs.proxy_type)
|
||||
tox_options.contents.proxy_host = bytes(oArgs.proxy_host, 'UTF-8')
|
||||
tox_options.contents.proxy_port = int(oArgs.proxy_port)
|
||||
tox_options.contents.udp_enabled = False
|
||||
else:
|
||||
tox_options.contents.udp_enabled = oArgs.udp_enabled
|
||||
if not os.path.exists('/proc/sys/net/ipv6'):
|
||||
oArgs.ipv6_enabled = False
|
||||
else:
|
||||
tox_options.contents.ipv6_enabled = oArgs.ipv6_enabled
|
||||
|
||||
tox_options.contents.tcp_port = int(oArgs.tcp_port)
|
||||
tox_options.contents.dht_announcements_enabled = oArgs.dht_announcements_enabled
|
||||
tox_options.contents.hole_punching_enabled = oArgs.hole_punching_enabled
|
||||
|
||||
# overrides
|
||||
tox_options.contents.local_discovery_enabled = False
|
||||
tox_options.contents.experimental_thread_safety = False
|
||||
# REQUIRED!!
|
||||
if oArgs.ipv6_enabled and not os.path.exists('/proc/sys/net/ipv6'):
|
||||
LOG.warning('Disabling IPV6 because /proc/sys/net/ipv6 does not exist' + repr(oArgs.ipv6_enabled))
|
||||
tox_options.contents.ipv6_enabled = False
|
||||
else:
|
||||
tox_options.contents.ipv6_enabled = bool(oArgs.ipv6_enabled)
|
||||
|
||||
if data: # load existing profile
|
||||
tox_options.contents.savedata_type = enums.TOX_SAVEDATA_TYPE['TOX_SAVE']
|
||||
tox_options.contents.savedata_data = c_char_p(data)
|
||||
tox_options.contents.savedata_length = len(data)
|
||||
else: # create new profile
|
||||
tox_options.contents.savedata_type = enums.TOX_SAVEDATA_TYPE['NONE']
|
||||
tox_options.contents.savedata_data = None
|
||||
tox_options.contents.savedata_length = 0
|
||||
|
||||
#? tox_options.contents.log_callback = LOG
|
||||
if tox_options._options_pointer:
|
||||
# LOG.debug("Adding logging to tox_options._options_pointer ")
|
||||
vAddLoggerCallback(tox_options, on_log)
|
||||
else:
|
||||
LOG.warning("No tox_options._options_pointer " +repr(tox_options._options_pointer))
|
||||
|
||||
return tox_options
|
||||
|
||||
def oMainArgparser(_=None, iMode=0):
|
||||
# 'Mode: 0=chat 1=chat+audio 2=chat+audio+video default: 0'
|
||||
if not os.path.exists('/proc/sys/net/ipv6'):
|
||||
@ -769,7 +816,7 @@ def bootstrap_udp(lelts, lToxes, oArgs=None):
|
||||
continue
|
||||
if not oRet:
|
||||
LOG.warn(f'bootstrap_udp failed to {host} : {oRet}')
|
||||
elif oTox.self_get_connection_status() != TOX_CONNECTION['NONE']:
|
||||
elif oTox.self_get_connection_status() != enums.TOX_CONNECTION['NONE']:
|
||||
LOG.info(f'bootstrap_udp to {host} connected')
|
||||
break
|
||||
else:
|
||||
@ -813,7 +860,7 @@ def bootstrap_tcp(lelts, lToxes, oArgs=None):
|
||||
elif hasattr(oTox, 'mycon_status') and oTox.mycon_status is False:
|
||||
LOG.info(f'bootstrap_tcp to {host} not True' \
|
||||
+f" last={int(oTox.mycon_time)}" )
|
||||
elif oTox.self_get_connection_status() != TOX_CONNECTION['NONE']:
|
||||
elif oTox.self_get_connection_status() != enums.TOX_CONNECTION['NONE']:
|
||||
LOG.info(f'bootstrap_tcp to {host} connected' )
|
||||
break
|
||||
else:
|
||||
@ -863,7 +910,7 @@ def iNmapInfo(sProt, sHost, sPort, key=None, environ=None, cmd='nmap'):
|
||||
return 0
|
||||
|
||||
|
||||
# ts.bootstrap_iNmapInfo(lElts, self._args, sProt)
|
||||
# bootstrap_iNmapInfo(lElts, self._args, sProt)
|
||||
def bootstrap_iNmapInfo(lElts, oArgs, protocol="tcp4", bIS_LOCAL=False, iNODES=iNODES, cmd='nmap'):
|
||||
if not bIS_LOCAL and not bAreWeConnected():
|
||||
LOG.warn(f"bootstrap_iNmapInfo not local and NOT CONNECTED")
|
||||
|
@ -1588,7 +1588,7 @@ class ToxSuite(unittest.TestCase):
|
||||
if hasattr(self, 'baid') and self.baid >= 0:
|
||||
self.bob.friend_delete(self.baid)
|
||||
|
||||
#? @unittest.skip('crashes')
|
||||
@unittest.skip('crashes')
|
||||
def test_kill_remake(self):
|
||||
"""
|
||||
t:friend_get_kill_remake
|
||||
@ -2078,7 +2078,7 @@ class ToxSuite(unittest.TestCase):
|
||||
|
||||
LOG_INFO(f"test_file_transfer:: self.wait_objs_attr completed")
|
||||
|
||||
@unittest.skip('crashes')
|
||||
#? @unittest.skip('crashes')
|
||||
def test_tox_savedata(self): # works sorta
|
||||
"""
|
||||
t:get_savedata_size
|
||||
@ -2098,6 +2098,7 @@ class ToxSuite(unittest.TestCase):
|
||||
LOG.info("test_tox_savedata alice.kill")
|
||||
# crashes
|
||||
self.alice.kill()
|
||||
del self.alice
|
||||
except:
|
||||
pass
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user