DMRlink/record.py

103 lines
4.0 KiB
Python
Raw Normal View History

2014-08-24 18:14:20 -04:00
#!/usr/bin/env python
#
# This work is licensed under the Creative Commons Attribution-ShareAlike
# 3.0 Unported License.To view a copy of this license, visit
# http://creativecommons.org/licenses/by-sa/3.0/ or send a letter to
# Creative Commons, 444 Castro Street, Suite 900, Mountain View,
# California, 94041, USA.
# This is a sample application that "records" voice transmissions to
# a datafile... presumably to be played back later.
from __future__ import print_function
from twisted.internet import reactor
from binascii import b2a_hex as h
import sys
2014-08-24 20:01:41 -04:00
import cPickle as pickle
2014-08-24 18:14:20 -04:00
from dmrlink import IPSC, NETWORK, networks, logger, int_id, hex_str_3
__author__ = 'Cortney T. Buffington, N0MJS'
__copyright__ = 'Copyright (c) 2014 Cortney T. Buffington, N0MJS and the K0USY Group'
__credits__ = 'Adam Fast, KC0YLK; Dave K; and he who wishes not to be named'
__license__ = 'Creative Commons Attribution-ShareAlike 3.0 Unported'
__maintainer__ = 'Cort Buffington, N0MJS'
__email__ = 'n0mjs@me.com'
__status__ = 'pre-alpha'
print('This program will record the first matching voice call and exit.\n')
while True:
2014-08-24 22:03:08 -04:00
tx_type = raw_input('Group (g) or Private voice (p)? ')
if tx_type == 'g' or tx_type == 'p':
2014-08-24 18:14:20 -04:00
break
print('...input must be either \'g\' or \'p\'')
while True:
2014-08-24 22:03:08 -04:00
ts = raw_input('Which timeslot (1, 2 or \'both\')? ')
if ts == '1' or ts == '2' or ts =='both':
if ts == '1':
ts = (0,)
if ts == '2':
ts = (1,)
if ts == 'both':
ts = (0,1)
2014-08-24 18:14:20 -04:00
break
print('...input must be \'1\', \'2\' or \'both\'')
2014-08-24 22:03:08 -04:00
id = raw_input('Which Group or Subscriber ID to record? ')
id = int(id)
id = hex_str_3(id)
2014-08-24 18:14:20 -04:00
2014-08-24 22:03:08 -04:00
filename = raw_input('Filename to use for this recording? ')
2014-08-24 18:14:20 -04:00
class recordIPSC(IPSC):
def __init__(self, *args, **kwargs):
IPSC.__init__(self, *args, **kwargs)
self.CALL_DATA = []
#************************************************
# CALLBACK FUNCTIONS FOR USER PACKET TYPES
#************************************************
#
2014-08-24 22:03:08 -04:00
if tx_type == 'g':
2014-08-24 18:14:20 -04:00
print('Initializing to record GROUP VOICE transmission')
def group_voice(self, _network, _src_sub, _dst_sub, _ts, _end, _peerid, _data):
2014-08-24 22:03:08 -04:00
if id == _dst_sub and _ts in ts:
2014-08-24 18:14:20 -04:00
if not _end:
if not self.CALL_DATA:
print('({}) Recording transmission from subscriber: {}' .format(_network, int_id(_src_sub)))
self.CALL_DATA.append(_data)
if _end:
self.CALL_DATA.append(_data)
2014-08-24 22:03:08 -04:00
print('({}) Transmission ended, writing to disk: {}' .format(_network, filename))
pickle.dump(self.CALL_DATA, open(filename, 'wb'))
2014-08-24 18:14:20 -04:00
reactor.stop()
print('Recording created, program terminating')
2014-08-24 22:03:08 -04:00
if tx_type == 'p':
2014-08-24 18:14:20 -04:00
print('Initializing ro record PRIVATE VOICE transmission')
def private_voice(self, _network, _src_sub, _dst_sub, _ts, _end, _peerid, _data):
2014-08-24 22:03:08 -04:00
if id == _dst_sub and _ts in ts:
2014-08-24 18:14:20 -04:00
if not _end:
if not self.CALL_DATA:
print('({}) Recording transmission from subscriber: {}' .format(_network, int_id(_src_sub)))
self.CALL_DATA.append(_data)
if _end:
self.CALL_DATA.append(_data)
2014-08-24 22:03:08 -04:00
print('({}) Transmission ended, writing to disk: {}' .format(_network, filename))
pickle.dump(self.CALL_DATA, open(filename, 'wb'))
2014-08-24 18:14:20 -04:00
reactor.stop()
print('Recording created, program terminating')
if __name__ == '__main__':
logger.info('DMRlink \'record.py\' (c) 2014 N0MJS & the K0USY Group - SYSTEM STARTING...')
for ipsc_network in NETWORK:
if NETWORK[ipsc_network]['LOCAL']['ENABLED']:
networks[ipsc_network] = recordIPSC(ipsc_network)
reactor.listenUDP(NETWORK[ipsc_network]['LOCAL']['PORT'], networks[ipsc_network])
reactor.run()