207 Commits

Author SHA1 Message Date
Logan Cusano
b01ec700f7 Merge branch 'master' into Noisegate_sounddevice_update 2022-04-23 01:46:34 -04:00
b2a73726a0 Merge pull request 'V3_GQRX_Improvements' (#11) from V3_GQRX_Improvements into master
Reviewed-on: http://git.vpn.cusano.net/Discord_Bot_Gang/Discord-Radio-Bot/pulls/11
2022-04-16 20:52:43 -04:00
Logan Cusano
a094ba581d //WIP v3
GQRX Changes
- Better error handling; GQRX config dupes some options for an unknown reason
2022-04-10 13:53:59 -04:00
Logan Cusano
cd69fa69ed //WIP Sounddevice migration
Noisegate - Incorrect kwarg name
2022-04-10 00:04:29 -04:00
Logan Cusano
3d73da72d6 //WIP Sounddevice migration
Bot bug - Incorrect kwarg name
2022-04-10 00:03:28 -04:00
Logan Cusano
663735210c //WIP Sounddevice migration
Incorrect value set for 'hostapi'
2022-04-10 00:01:32 -04:00
Logan Cusano
4371b02ab6 //WIP Sounddevice migration
Bot bug fix
2022-04-09 23:57:47 -04:00
Logan Cusano
80f83f2026 //WIP Sounddevice migration
Initial
2022-04-09 23:54:28 -04:00
Logan Cusano
62f9e1abaa //WIP v3
GQRX Changes
- Ensure AGC is enabled whenever the bot starts
2022-04-09 22:34:56 -04:00
Logan Cusano
e4e04e429f //WIP v3
GQRX Config Template
- Added switch to allow AGC
2022-04-09 22:27:26 -04:00
Logan Cusano
0db4f62afc //WIP v3
Clear Messages Changes
- Bug
- More error checking
- Limit the user to 100 messages
- More help text
2022-04-09 22:22:44 -04:00
Logan Cusano
4d5c1183d6 //WIP v3
Clear Messages Changes
- Bug
2022-04-09 22:15:37 -04:00
Logan Cusano
9dba89a60b //WIP v3
Clear Messages Changes
- Check for negative numbers... Gino
2022-04-09 22:14:21 -04:00
Logan Cusano
dcf0de8351 //WIP v3
Clear Messages Changes
- Added help message
2022-04-09 22:09:24 -04:00
Logan Cusano
bfb049e268 //WIP v3
Clear Messages Changes
- Default clear to 2 messages (instantiation and the previous message)
2022-04-09 22:05:32 -04:00
Logan Cusano
b0d83def35 //WIP v3
Clear Messages Changes
- No "#0000" in username output
2022-04-09 22:03:52 -04:00
Logan Cusano
8af7a70b65 //WIP v3
Clear Messages Changes
- Mentioning user ID not name
2022-04-09 22:02:04 -04:00
Logan Cusano
8d090c5c67 //WIP v3
Clear Messages Changes
- Better logging
- Better output
2022-04-09 21:58:34 -04:00
Logan Cusano
35e19ee83f //WIP v3
Clear Messages Changes
- New approach
2022-04-09 21:39:44 -04:00
Logan Cusano
44d027ab8c Merge branch 'master' into V3_GQRX_Improvements 2022-04-09 21:29:12 -04:00
Logan Cusano
31d5f8139c New Module to Clear Channel Messages 2022-04-09 21:28:21 -04:00
Logan Cusano
3837f7abd6 //WIP V3
Update todo
2022-04-09 21:27:49 -04:00
Logan Cusano
b7f0939787 //WIP V3
GQRX Changes
- Sending output to devnull, except for error
2022-04-09 04:32:12 -04:00
Logan Cusano
cccd974adc //WIP V3
GQRX Changes
- Changed repeating log entry to debug
2022-04-09 03:39:45 -04:00
Logan Cusano
f4b14c26fc Merge branch 'master' into V3_GQRX_Improvements 2022-04-09 03:27:23 -04:00
Logan Cusano
a483f83b4b //WIP V3
Bot Changes
- Increased loop delay when waiting for GQRX to start
Noisegate Change
- Display whether the noisegate is active or not in the logs
2022-04-09 02:36:48 -04:00
Logan Cusano
458ab5be95 //WIP V3
Bot Changes
- Never used new radio config function when loading config
2022-04-09 02:25:46 -04:00
Logan Cusano
dbbd5b6fe3 //WIP V3
Bot Changes
- Awaiting new async 'use_current_radio_config' function
2022-04-09 02:23:46 -04:00
Logan Cusano
63b37d14cb //WIP V3
Bot Changes
- Improved handlers
- Added new function to use the newly set or loaded settings
- Any change to frequency, squelch, mode, or noisegate threshold will show no profile
    - The user is responsible for saving their changes

Readme Changes
- Update with notes on the bot change above
2022-04-09 02:21:53 -04:00
Logan Cusano
2dabf1c98d //WIP V3
GQRX Changes
- Resetting tel_conn when closing GQRX
2022-04-09 01:57:24 -04:00
Logan Cusano
5551aa6355 //WIP V3
Default Radio Settings Update
- One location for default radio settings

GQRX Changes
- Default settings
- Simpler check to see if GQRX is started
- Working process handler
2022-04-09 01:52:37 -04:00
Logan Cusano
7ee5d92852 forgot each item needed to be in the list, no spaces 2022-04-09 01:21:33 -04:00
Logan Cusano
5a3729627b GQRX Changes
- Checks for GQRX config file
    - If one exists, reset the 'crashed' option
    - If one does not, create one from the template
- New templates package
    - Useful for storing large files to be called on later
2022-04-09 01:13:06 -04:00
Logan Cusano
e4201a2746 Fixed Logging 2022-04-08 21:46:28 -04:00
Logan Cusano
bb56566d37 Merge branch 'master' into V3_GQRX_Improvements 2022-04-08 21:35:07 -04:00
Logan Cusano
35e90aa290 Created a 'debug' log file to keep one log file clean 2022-04-08 21:32:31 -04:00
Logan Cusano
99ec1d07aa Merge remote-tracking branch 'origin/V3_GQRX_Improvements' into V3_GQRX_Improvements 2022-04-01 01:59:51 -04:00
Logan Cusano
bbca25ce3f Getting closer on the GQRX Handler. Doesn't close properly 2022-04-01 01:55:14 -04:00
Logan Cusano
676936eb6f More intelligent init of the handler 2022-04-01 01:10:12 -04:00
Logan Cusano
8debd690f1 Better logging 2022-04-01 01:08:54 -04:00
Logan Cusano
5d1a58ee87 Var got set before the error check 2022-04-01 01:07:23 -04:00
Logan Cusano
fd62da0c6f Silly bug 2022-04-01 01:05:57 -04:00
Logan Cusano
b972d09a65 Handling the delay in GQRX opening and being 'usable'
Handling the delay in GQRX opening and being 'usable'
2022-04-01 01:00:32 -04:00
Logan Cusano
21aacde130 Moved the main inits to when the bot is run 2022-04-01 00:51:34 -04:00
Logan Cusano
e26ae52399 Wrong location of the logger 2022-04-01 00:50:28 -04:00
Logan Cusano
319de5cf42 Initial changes to handle GQRX process 2022-04-01 00:46:34 -04:00
5e6b6f02d9 Merge pull request 'New logging system' (#10) from V3_Better_Logging into master
Reviewed-on: http://git.vpn.cusano.net/Discord_Bot_Gang/Discord-Radio-Bot/pulls/10
2022-04-01 00:02:08 -04:00
Logan Cusano
25ad3b076c Wait for the bot to connect before playing audio 2022-03-31 22:52:10 -04:00
Logan Cusano
d3a18c2421 BUG Would try to join when the bot was disconnected 2022-03-31 18:23:45 -04:00
Logan Cusano
6170719233 Update todo 2022-03-31 03:00:58 -04:00
Logan Cusano
c8503c856e Update todo with GQRX commands 2022-03-31 02:47:07 -04:00
Logan Cusano
471f668cdb Flipped the handlers 2022-03-31 02:32:14 -04:00
Logan Cusano
579987d571 Forgot to update new var name 2022-03-31 02:30:46 -04:00
Logan Cusano
6efca8f0fa Added a callback to itself so if it cuts out, it should restart 2022-03-31 02:28:54 -04:00
Logan Cusano
9e60444bc6 More specific log output 2022-03-31 02:27:21 -04:00
Logan Cusano
9280278ca7 Ignore the new ./logs directory 2022-03-30 01:38:22 -04:00
Logan Cusano
c07220f434 Update readme 2022-03-30 01:37:54 -04:00
Logan Cusano
d650ae4350 Never updated the global sensitivity, so the profile would be updated with this value 2022-03-30 01:35:43 -04:00
Logan Cusano
cbeb90a0f4 Remove closing 2022-03-30 01:31:19 -04:00
Logan Cusano
5c0d0f1f7b Merge remote-tracking branch 'origin/V3_Better_Logging' into V3_Better_Logging 2022-03-30 00:37:17 -04:00
Logan Cusano
6135bd81de Specified output name
Specified output name
2022-03-30 00:37:01 -04:00
Logan Cusano
dc178e54af Specified output name 2022-03-30 00:33:13 -04:00
Logan Cusano
f124853332 Specified output name 2022-03-30 00:20:36 -04:00
Logan Cusano
331753ccbf Merge remote-tracking branch 'origin/V3_Better_Logging' into V3_Better_Logging 2022-03-29 23:39:28 -04:00
Logan Cusano
fd8fb1ab3d undoing verbosity change 2022-03-29 23:36:46 -04:00
Logan Cusano
5b2d0f9135 Merge branch 'V3_Better_Logging' of git.vpn.cusano.net:Discord_Bot_Gang/Discord-Radio-Bot into V3_Better_Logging 2022-03-29 23:31:13 -04:00
Logan Cusano
1357a0509c Merge branch 'V3_Better_Logging' of git.vpn.cusano.net:Discord_Bot_Gang/Discord-Radio-Bot into V3_Better_Logging 2022-03-29 23:20:44 -04:00
Logan Cusano
ac4e4349df OCD change 2022-03-30 03:17:37 +00:00
Logan Cusano
9af6fc254c Added logs to a folder, so it doesn't polute the root 2022-03-30 03:17:37 +00:00
Logan Cusano
c1ecd092c6 Added logging 2022-03-30 03:17:36 +00:00
Logan Cusano
34c5fe9c5b Increase verbosity 2022-03-30 03:17:36 +00:00
Logan Cusano
65b0f25921 New logging 2022-03-30 03:17:36 +00:00
a8fbd094e1 Update '.gitignore'
Better directory management
2022-03-29 23:15:14 -04:00
Logan Cusano
834a74c7f0 Merge branch 'V3_Better_Logging' of git.vpn.cusano.net:Discord_Bot_Gang/Discord-Radio-Bot into V3_Better_Logging 2022-03-29 23:13:09 -04:00
Logan Cusano
d2bf7822cf OCD change 2022-03-29 23:08:40 -04:00
Logan Cusano
22d36c5f19 Added logs to a folder, so it doesn't polute the root 2022-03-29 23:06:43 -04:00
Logan Cusano
a9cfa89655 Added logging 2022-03-29 22:56:20 -04:00
Logan Cusano
44e0941a6a Increase verbosity 2022-03-29 22:42:16 -04:00
Logan Cusano
e6d4c9f467 Merge remote-tracking branch 'origin/V3_Better_Logging' into V3_Better_Logging 2022-03-29 22:39:25 -04:00
Logan Cusano
acc76db5f9 Merge branch 'V3_Better_Logging' of git.vpn.cusano.net:Discord_Bot_Gang/Discord-Radio-Bot into V3_Better_Logging 2022-03-29 16:45:03 -04:00
Logan Cusano
1ed5436138 New logging 2022-03-29 20:44:52 +00:00
Logan Cusano
f8f6cb7e52 Stopping the voice connection after the fadeout and change fadeout duration 2022-03-29 16:39:48 -04:00
Logan Cusano
458d8d12a4 Merge branch 'V3_Better_Logging' of git.vpn.cusano.net:Discord_Bot_Gang/Discord-Radio-Bot into V3_Better_Logging 2022-03-28 01:58:23 -04:00
Logan Cusano
319e92b897 New logging 2022-03-28 05:56:20 +00:00
Logan Cusano
997c24949b Updated todo 2022-03-28 01:55:30 -04:00
Logan Cusano
326da757cf No pipes or STDIN/STDOUT 2022-03-28 01:46:50 -04:00
Logan Cusano
c741288b87 New logging 2022-03-28 01:27:37 -04:00
d97e51d960 Merge pull request 'Noisegate V2' (#8) from NoiseGateV2 into master
Reviewed-on: http://git.vpn.cusano.net/Discord_Bot_Gang/Discord-Radio-Bot/pulls/8
2022-03-28 00:13:06 -04:00
Logan Cusano
8c5cc65a50 no new line at the end of display profiles 2022-03-27 15:05:46 -04:00
Logan Cusano
63ca0fec63 silly bug 2022-03-27 15:04:55 -04:00
Logan Cusano
8e191531d0 Reads profiles without noisegate sensitivity value 2022-03-27 15:03:41 -04:00
Logan Cusano
83c2971742 No spaces in profiles 2022-03-27 15:00:58 -04:00
Logan Cusano
1a705c2b6c KeyError instead of NameError ~507 bot.py 2022-03-27 14:56:56 -04:00
Logan Cusano
3a2d9531f9 Removing circular import 2022-03-27 14:55:11 -04:00
Logan Cusano
54e3a3a116 Remove dupe file 2022-03-27 14:50:42 -04:00
Logan Cusano
0d32d3fa7e rebase master into NoiseGateV2 2022-03-27 14:48:42 -04:00
Logan Cusano
05e81aea1d Better output 2022-03-27 14:48:40 -04:00
Logan Cusano
69c7d4d548 idk 2022-03-27 14:48:39 -04:00
Logan Cusano
79c651be30 idk 2022-03-27 14:48:38 -04:00
Logan Cusano
4eb73c3832 idk 2022-03-27 14:48:37 -04:00
Logan Cusano
f90c84184a idk 2022-03-27 14:48:36 -04:00
Logan Cusano
831c19e7e1 idk 2022-03-27 14:48:34 -04:00
Logan Cusano
97aed314ba idk 2022-03-27 14:48:33 -04:00
Logan Cusano
f64a3dc517 idk 2022-03-27 14:48:32 -04:00
Logan Cusano
5d7a3b2d0b idk 2022-03-27 14:48:31 -04:00
Logan Cusano
0c53e58fab idk 2022-03-27 14:48:30 -04:00
Logan Cusano
8c7075b76c idk 2022-03-27 14:48:28 -04:00
Logan Cusano
dd9643363b idk 2022-03-27 14:48:27 -04:00
Logan Cusano
7cca55a836 Trying to use pipe 2022-03-27 14:48:26 -04:00
Logan Cusano
5fe98bfe0e Trying to use stdin, executable and shell 2022-03-27 14:48:25 -04:00
Logan Cusano
6049af3a25 Trying to use stdout/stderr 2022-03-27 14:48:24 -04:00
Logan Cusano
c03521de7f Using a list with extend instead of append 2022-03-27 14:48:22 -04:00
Logan Cusano
e5b3359571 Using a list with extend instead of append 2022-03-27 14:48:21 -04:00
Logan Cusano
492026ed8c Using a list with extend instead of append 2022-03-27 14:48:20 -04:00
Logan Cusano
61f113c475 Using a list with extend instead of append 2022-03-27 14:48:19 -04:00
Logan Cusano
bbe461255a idk 2022-03-27 14:48:18 -04:00
Logan Cusano
7d161baeda no space in debug emendation 2022-03-27 14:48:16 -04:00
Logan Cusano
f68e794d02 popen needs string? 2022-03-27 14:48:15 -04:00
Logan Cusano
9986eadaaf popen needs string? 2022-03-27 14:48:14 -04:00
Logan Cusano
404ddfc773 Updated p25 opening 2022-03-27 14:48:13 -04:00
Logan Cusano
f94a900c44 Added profile commands for the bot 2022-03-27 14:43:33 -04:00
Logan Cusano
721ef932ce Better output 2022-03-27 14:18:16 -04:00
Logan Cusano
4ab14d2e5d idk 2022-03-27 14:08:53 -04:00
Logan Cusano
9635e11434 idk 2022-03-27 14:06:36 -04:00
Logan Cusano
8fa42d819a idk 2022-03-27 14:03:40 -04:00
Logan Cusano
5a8efa8884 idk 2022-03-27 14:01:49 -04:00
Logan Cusano
e0dba3be4e idk 2022-03-27 13:52:37 -04:00
Logan Cusano
4e839eecb4 idk 2022-03-27 13:50:24 -04:00
Logan Cusano
81dc305197 idk 2022-03-27 13:49:02 -04:00
Logan Cusano
203bee0734 idk 2022-03-27 13:48:07 -04:00
Logan Cusano
7e2cf932dc idk 2022-03-27 13:46:44 -04:00
Logan Cusano
102e0ae473 idk 2022-03-27 13:43:52 -04:00
Logan Cusano
33ea596ab4 idk 2022-03-27 13:40:00 -04:00
Logan Cusano
f965c5de05 Trying to use pipe 2022-03-27 13:32:49 -04:00
Logan Cusano
0bc0d8e031 Trying to use stdin, executable and shell 2022-03-27 13:29:15 -04:00
Logan Cusano
92ab4992cb Trying to use stdout/stderr 2022-03-27 13:19:29 -04:00
Logan Cusano
da29c0eaaa Using a list with extend instead of append 2022-03-27 13:12:25 -04:00
Logan Cusano
30d539312e Using a list with extend instead of append 2022-03-27 13:03:00 -04:00
Logan Cusano
9a225010be Using a list with extend instead of append 2022-03-27 13:00:31 -04:00
Logan Cusano
92bd4ffb1f Using a list with extend instead of append 2022-03-27 12:49:08 -04:00
Logan Cusano
bf8adda80b idk 2022-03-27 05:20:00 -04:00
Logan Cusano
f239e87b0a no space in debug emendation 2022-03-27 05:18:23 -04:00
Logan Cusano
136009c07e popen needs string? 2022-03-27 05:17:03 -04:00
Logan Cusano
5efb44164a popen needs string? 2022-03-27 05:11:49 -04:00
Logan Cusano
6fd30b9609 Updated p25 opening 2022-03-27 04:37:29 -04:00
Logan Cusano
c1084deadf Update requirements from Dan's Update 2022-03-27 01:24:36 -04:00
Logan Cusano
cf79050b96 Possible fix for Nonetype error on Greada 2022-03-27 01:16:53 -04:00
Logan Cusano
22d6d5d3d7 Added a function to allow users to change the noisegate threshold 2022-03-27 01:15:17 -04:00
Logan Cusano
24adc47109 Dan's Update
- New classes for noisegate & audio stream
- Ability to manipulate raw data in real-time
2022-03-27 00:50:42 -04:00
Logan Cusano
0329ff7a81 New attempt 2022-03-26 03:07:24 -04:00
Logan Cusano
48f6812f68 Printing volume value 2022-03-26 02:42:42 -04:00
Logan Cusano
b444236dc9 Printing volume value 2022-03-26 02:39:08 -04:00
Logan Cusano
287eb0e981 Merge remote-tracking branch 'origin/NoiseGateV2' into NoiseGateV2 2022-03-26 02:34:39 -04:00
Logan Cusano
7a9dbfb53a Actually connecting the noisegate 2022-03-26 02:34:01 -04:00
Logan Cusano
a13cf36e15 Actually connecting the noisegate 2022-03-26 02:24:06 -04:00
Logan Cusano
465c0e00ea Added noisegate to the 'sound' file and integrated it with its function 2022-03-26 01:35:02 -04:00
Logan Cusano
12aa4d6f51 Merge branch 'master' into NoiseGateV2 2022-03-20 03:15:37 -04:00
Logan Cusano
2dacab263b Rename to resolve conflict 2022-03-20 03:05:33 -04:00
Logan Cusano
126ba907af Testing OP25 Improvements
- New system of handling thread
2022-03-19 01:18:45 -04:00
Logan Cusano
4f230f8d33 Testing OP25 Improvements
- Introduced threading (again..)
2022-03-19 00:46:13 -04:00
Logan Cusano
1257b713fc Testing OP25 Improvements
- Disabled shell
2022-03-19 00:39:10 -04:00
Logan Cusano
4f7ec14807 Testing OP25 Improvements
- Changed output to be ignored
2022-03-19 00:32:03 -04:00
Logan Cusano
ec7880ed1e Testing OP25 Improvements 2022-03-19 00:27:55 -04:00
Logan Cusano
e4cb8647ee Replaced arbitrary sleep with a wait until response on all functions 2022-03-19 00:19:02 -04:00
Logan Cusano
a205bd2d8a BUG; Used string, not byte string 2022-03-19 00:17:36 -04:00
Logan Cusano
8355b98da5 Replaced arbitrary sleep with a wait until response 2022-03-19 00:16:24 -04:00
Logan Cusano
c3fb20f12e Increased sleep time 2022-03-19 00:07:06 -04:00
Logan Cusano
5ebbaca54b GQRX bug; Never created connection 2022-03-17 23:30:40 -04:00
Logan Cusano
efe8eda66f Improved GQRX handler speed 2022-03-17 23:28:59 -04:00
Logan Cusano
9f4473ffcc Improved the display of profiles 2022-03-17 23:21:29 -04:00
Logan Cusano
759f05eb0c Improved the display of profiles 2022-03-17 23:18:17 -04:00
Logan Cusano
0d51b47146 BUGFIX in display all profiles 2022-03-17 23:15:02 -04:00
Logan Cusano
ff26cbc0b1 Removed ping in favor of 'ModuleProbe' module 2022-03-17 23:13:07 -04:00
Logan Cusano
9456b91d09 Added WYA command to display where the bot is 2022-03-17 23:10:13 -04:00
Logan Cusano
b534e2b731 Typo 2022-03-17 22:41:37 -04:00
fdeb337ca5 Merge branch 'NoiseGateV2' into master 2022-03-17 16:52:12 -04:00
Logan Cusano
56a3bdd0ea Default the OP25 HTTP server to enabled, caused issues with the shell with it disabled.
Need to look into perhaps using the `shell=True` parameter
2022-03-17 00:12:15 -04:00
Logan Cusano
830d299a01 Update gitignore to ignore all pycache folders 2022-03-17 00:09:39 -04:00
676969a8fe Merge pull request 'Moved the startup ping response to a module, so it will only reply if the modules are enabled' (#6) from ModuleProbe into master
Reviewed-on: http://git.vpn.cusano.net/Discord_Bot_Gang/Discord-Radio-Bot/pulls/6
2022-03-17 00:00:46 -04:00
Logan Cusano
760ebd7e43 Added a call to set squelch to 0 as all settings are being changed, so there is no sudden burst of noise 2022-03-16 23:59:42 -04:00
Logan Cusano
c564a145b5 Moved the startup ping response to a module, so it will only reply if the modules are enabled 2022-03-16 23:52:17 -04:00
Logan Cusano
db0922bf68 Updated TODO 2022-03-16 02:20:47 -04:00
Logan Cusano
897e53d4c0 Changed the order of operations on boot so squelch is changed first 2022-03-16 02:20:35 -04:00
Logan Cusano
a1f9869e8e Undo revert 2022-03-16 01:34:02 -04:00
Logan Cusano
73fa560040 Removed noisegate (again) 2022-03-16 01:29:09 -04:00
Logan Cusano
1f30e05e46 Revert "Cleanup"
This reverts commit 78d9b9e307.
2022-03-16 01:26:24 -04:00
Logan Cusano
56ee57e96a Merge remote-tracking branch 'origin/master' 2022-03-16 01:15:47 -04:00
Logan Cusano
5cc8a8c118 Changed to sync sleep 2022-03-16 01:15:22 -04:00
26909bce0f Merge pull request 'Merge from master' (#5) from master into NoiseGateV2
Reviewed-on: http://git.vpn.cusano.net/Discord_Bot_Gang/Discord-Radio-Bot/pulls/5
2022-02-27 21:47:07 -05:00
fc1a742933 Merge branch 'NoiseGateV2' into master 2022-02-27 21:46:31 -05:00
Logan Cusano
78d9b9e307 Cleanup 2022-02-27 21:43:28 -05:00
Logan Cusano
ba237d2d53 Working on bug in leaving 2022-02-27 19:09:48 -05:00
Logan Cusano
6ad489183f Working on bug in leaving 2022-02-27 19:01:52 -05:00
Logan Cusano
b721fa3df4 Working on bug in leaving 2022-02-27 19:00:55 -05:00
Logan Cusano
7682d8c72c Working on bug in leaving 2022-02-27 18:44:02 -05:00
Logan Cusano
706f4a31df Working on bug in leaving 2022-02-27 18:40:36 -05:00
Logan Cusano
7fa11a4abe Working on bug in leaving 2022-02-27 18:36:18 -05:00
Logan Cusano
17c528c153 Working on bug in leaving 2022-02-27 17:29:08 -05:00
Logan Cusano
f9594e2afe Error catching
- PCMStream.clean_up
- PCMStream.read
2022-02-27 17:25:10 -05:00
Logan Cusano
167c06f331 Error catching
- PCMStream.clean_up
- PCMStream.read
2022-02-27 17:21:53 -05:00
Logan Cusano
137d0ae097 Error catching for cleaning up 2022-02-27 17:17:23 -05:00
Logan Cusano
002288d632 Error catching for cleaning up 2022-02-27 17:14:31 -05:00
Logan Cusano
59a3ca22ea Debug output for leaving 2022-02-27 17:11:59 -05:00
Logan Cusano
fdb9bb519c Grammar fix
Comma in `Ok <member>*,*\n<message>`
2022-02-27 17:09:06 -05:00
Logan Cusano
a5f0a4b193 BUGFIX appending to dict, fixed 2022-02-27 16:30:38 -05:00
Logan Cusano
62968654d2 v2.0.1
- Args
- Better configs
- Other shit too, idr
2022-02-27 13:40:21 -05:00
Logan Cusano
a3a592d854 New Noisegate attempt 2022-02-14 20:13:23 -05:00
19 changed files with 1084 additions and 372 deletions

8
.gitignore vendored
View File

@@ -1,12 +1,12 @@
/.idea/
/Releases/
/Old/
/__pycache__/
*/__pycache__/
/venv/
**/__pycache__/
**/venv/
*.7z
*.bat
/DSDPlus/
._.DS_Store
**/._.DS_Store
*.log
*.ini
**/logs/*

View File

@@ -1,9 +1,40 @@
import sound
import configparser
import logging
import os
from datetime import date
from os.path import exists
from NoiseGatev2 import AudioStream, query_devices
PDB_ACCEPTABLE_HANDLERS = ['gqrx', 'op25']
PDB_KNOWN_BOT_IDS = [756327271597473863, 915064996994633729, 943742040255115304]
# Handler configs
PDB_ACCEPTABLE_HANDLERS = {'gqrx': {
'Modes': ['wfm', 'fm']
},
'op25': {
'Modes': ['d', 'p25']
}}
# Known bot IDs
PDB_KNOWN_BOT_IDS = {756327271597473863: "Greada", 915064996994633729: "Jorn", 943742040255115304: "Brent"}
# Default value to set noisegate on new or unknown profiles
DEFAULT_NOISEGATE_THRESHOLD = 50
# Initialize the logger for this file
LOGGER = logging.getLogger('Discord_Radio_Bot.Bot_Resources')
# Location of the gqrx binary
GQRX_BIN_LOCATION = "/usr/bin/"
GQRX_BIN = "/usr/bin/gqrx"
# Default radio settings
DEFAULT_RADIO_SETTINGS = {
'profile_name': None,
'freq': "104700000",
'mode': "wfm",
'squelch': 0,
'noisegate_sensitivity': DEFAULT_NOISEGATE_THRESHOLD,
}
def check_if_config_exists():
@@ -31,9 +62,9 @@ def read_config_file():
'Handler': str(config['Config']['Handler'])
}
print("Found config options:")
LOGGER.debug("Found config options:")
for key in config_return.keys():
print(f"\t{key} : {config_return[key]}")
LOGGER.debug(f"\t{key} : {config_return[key]}")
return config_return
@@ -84,14 +115,14 @@ def write_config_file(**kwargs):
with open('./config.ini', 'w') as config_file:
config.write(config_file)
print("Config Saved")
LOGGER.info("Config Saved")
return True
def get_device_list():
list_of_devices = sound.query_devices().items()
print(list_of_devices)
list_of_devices = query_devices().items()
LOGGER.debug(list_of_devices)
return list_of_devices
@@ -99,38 +130,38 @@ def get_user_device_selection():
device_list = get_device_list()
org_device_list = []
for device, dev_id in device_list:
print(f"{dev_id + 1}\t-\t{device}")
LOGGER.debug(f"{dev_id + 1}\t-\t{device}")
org_device_list.append((dev_id, device))
selected_list_id = None
selected_device = None
while not selected_list_id:
print(f"selected device: {selected_list_id}")
print(device_list)
LOGGER.debug(f"selected device: {selected_list_id}")
LOGGER.debug(device_list)
try:
selected_list_id = int(input(f"Please select the input device from above:\t"))
except Exception as e:
print(e)
LOGGER.warning(e)
continue
if int(selected_list_id) < int(len(device_list)):
print("ford")
LOGGER.debug("Selected ID within range")
continue
elif selected_list_id > int(len(device_list)):
print("Out of range, try again...")
LOGGER.debug("Out of range, try again...")
selected_list_id = None
continue
else:
selected_list_id = None
print("Internal error, try again")
LOGGER.error("Internal error, try again")
continue
for dev_dict in org_device_list:
print(list(device_list)[selected_list_id-1][0])
LOGGER.debug(list(device_list)[selected_list_id-1][0])
if dev_dict[1] == list(device_list)[selected_list_id-1][0]:
selected_device = dev_dict
print(selected_device)
LOGGER.debug(selected_device)
return selected_device
@@ -142,7 +173,7 @@ def get_user_token():
if len(token) == 59:
return token
else:
print('Length error in token, please try again...')
LOGGER.error('Length error in token, please try again...')
token = None
continue
@@ -161,7 +192,7 @@ def get_user_mention_channel_id():
if len(str(channel_id)) == len('757379843792044102'):
return channel_id
else:
print("Length error in ID, please try again")
LOGGER.error("Length error in ID, please try again")
channel_id = None
continue
@@ -172,7 +203,8 @@ def get_handler():
handler = str(input(f"Please enter the name of the handler you would like to use:\t"))
if handler in PDB_ACCEPTABLE_HANDLERS:
return handler
elif handler == '':
elif handler == ['', "none"]:
handler = "None"
return handler
@@ -185,3 +217,56 @@ def check_negative(s):
return False
except ValueError:
return False
# Check if message is a ping request and respond even if it is a bot
async def check_and_reply_to_ping(bot, message):
if "check_modules" in message.content:
ctx = await bot.get_context(message)
await bot.invoke(ctx)
return True
else:
await bot.process_commands(message)
return False
# Create the logger
def init_global_logger(_verbose_level: str = "WARNING"):
numeric_log_level = getattr(logging, _verbose_level.upper(), None)
if not isinstance(numeric_log_level, int):
raise ValueError('Invalid log level: %s' % _verbose_level)
else:
# create logger
init_logger = logging.getLogger('Discord_Radio_Bot')
init_logger.setLevel(logging.DEBUG)
# create file handler which logs even debug messages
if not exists("./logs/"):
os.mkdir("./logs/")
fh = logging.FileHandler(f'./logs/DRB-{date.today()}.log')
fh.setLevel(logging.INFO)
fh_debug = logging.FileHandler(f'./logs/DRB-{date.today()}.DEBUG.log')
fh_debug.setLevel(logging.DEBUG)
# create terminal handler with a higher log level
th = logging.StreamHandler()
th.setLevel(numeric_log_level)
# create formatter and add it to the handlers
fh_debug_formatter = logging.Formatter('[%(asctime)s %(name)s->%(funcName)s():%(lineno)s] - %(levelname)s - %(message)s')
fh_formatter = logging.Formatter('[%(asctime)s %(name)s->%(funcName)s()] - %(levelname)s - %(message)s')
th_formatter = logging.Formatter('[%(asctime)s %(name)s->%(funcName)s()] - %(levelname)s - %(message)s')
fh_debug.setFormatter(fh_debug_formatter)
fh.setFormatter(fh_formatter)
th.setFormatter(th_formatter)
# add the handlers to the logger
init_logger.addHandler(fh)
init_logger.addHandler(th)
init_logger.addHandler(fh_debug)
#if __name__ is not 'main':
# init_global_logger()
# LOGGER = logging.getLogger('Discord_Radio_Bot')

View File

@@ -1,67 +0,0 @@
import threading
import time
import numpy as np
import discord
import sounddevice as sd
from threading import Thread, Event
noise_gate_trigger = 0 # Set this value for the trigger on the noise-gate
voice_connection = None
audio_stream = None
last_callback_value = 0
class NoiseGate(Thread):
def __init__(self, trigger_value: int = 700):
global noise_gate_trigger
super(NoiseGate, self).__init__()
self.stream = None
self.stop_NG = threading.Event()
self.NG_Started = threading.Event()
noise_gate_trigger = trigger_value
def init_stream(self, num, _voice_connection, _audio_stream):
global voice_connection, audio_stream
self.stream = sd.InputStream(device=num, callback=stream_callback)
voice_connection = _voice_connection
audio_stream = _audio_stream
def run(self) -> None:
self.NG_Started.set()
self.stream.start()
while not self.stop_NG.is_set():
#print("Main loop start")
if last_callback_value >= noise_gate_trigger:
trigger_noise_gate(True)
while last_callback_value >= noise_gate_trigger:
time.sleep(6)
trigger_noise_gate(False)
#print("Main loop end")
self.stream.stop()
self.stream.close()
self.NG_Started.clear()
voice_connection.stop()
print('Thread #%s stopped' % self.ident)
def stream_callback(indata, *args):
global last_callback_value
volume_normalization = np.linalg.norm(indata) * 10
last_callback_value = int(volume_normalization)
#print(f"Callback Value: {last_callback_value}")
def trigger_noise_gate(triggered: bool):
if triggered:
if not voice_connection.is_playing():
voice_connection.play(discord.PCMAudio(audio_stream))
# print("|" * int(volume_normalization / 4))
print("Noise Gate was Triggered")
#time.sleep(10)
else:
if voice_connection.is_playing():
print("Noise Gate stopped")
voice_connection.pause()
# try disconnecting and reconnecting

188
NoiseGatev2.py Normal file
View File

@@ -0,0 +1,188 @@
import audioop
import logging
import math
import time
import sounddevice
import pyaudio
import discord
import numpy
from pprint import pformat
voice_connection = None
LOGGER = logging.getLogger("Discord_Radio_Bot.NoiseGateV2")
DEFAULT = 0
# noinspection PyUnresolvedReferences
class AudioStream:
def __init__(self, _channels: int = 2, _sample_rate: int = 48000, _frames_per_buffer: int = 1024,
_device_index: int = None, _init_on_startup: bool = True):
self.sd_kwargs = {
'dtype': 'int16',
'channels': _channels,
'samplerate': _sample_rate,
'blocksize': _frames_per_buffer
}
if _device_index:
self.sd_kwargs['device'] = _device_index
# Define and initialize stream object if we have been passed a device ID (pyaudio.open)
self.stream = None
if _device_index:
if _init_on_startup:
LOGGER.info("Init stream")
self.init_stream()
def init_stream(self, _new_device_index: int = None):
# Check what device was asked to be changed (or set)
if _new_device_index:
self.sd_kwargs['device'] = _new_input_device_index
self.close_if_open()
# Open the stream
self.stream = sounddevice.RawStream(**self.sd_kwargs)
def close_if_open(self):
# Stop the stream if it is started
if self.stream:
if self.stream.active:
self.stream.stop()
self.stream.close()
LOGGER.debug(f"[ReopenStream.close_if_open]:\t Stream was open; It was closed.")
async def stop(self):
await voice_connection.disconnect()
self.close_if_open()
self.stream.close()
self.paInstance.terminate()
# noinspection PyUnresolvedReferences
class NoiseGate(AudioStream):
def __init__(self, _voice_connection, _noise_gate_threshold: int, **kwargs):
super(NoiseGate, self).__init__(_init_on_startup=True, **kwargs)
global voice_connection
voice_connection = _voice_connection
self.THRESHOLD = _noise_gate_threshold
self.NGStream = NoiseGateStream(self)
self.Voice_Connection_Thread = None
def run(self) -> None:
global voice_connection
# Start the audio stream
LOGGER.debug(f"Starting stream")
self.stream.start()
# Start the stream to discord
self.core()
def core(self, error=None):
if error:
LOGGER.warning(error)
while not voice_connection.is_connected():
time.sleep(.2)
if not voice_connection.is_playing():
LOGGER.debug(f"Playing stream to discord")
voice_connection.play(self.NGStream, after=self.core)
async def close(self):
LOGGER.debug(f"Closing")
await voice_connection.disconnect()
if self.stream.active:
self.stream.stop()
LOGGER.debug(f"Stopping stream")
# noinspection PyUnresolvedReferences
class NoiseGateStream(discord.AudioSource):
def __init__(self, _stream):
super(NoiseGateStream, self).__init__()
self.stream = _stream # The actual audio stream object
self.NG_fadeout = 240/20 # Fadeout value used to hold the noisegate after de-triggering
self.NG_fadeout_count = 0 # A count set when the noisegate is triggered and was de-triggered
self.process_set_count = 0 # Counts how many processes have been made
def read(self):
try:
while voice_connection.is_connected():
curr_buffer = bytearray(self.stream.stream.read(960))
buffer_rms = audioop.rms(curr_buffer, 2)
if buffer_rms > 0:
buffer_decibel = 20 * math.log10(buffer_rms)
if self.process_set_count % 10 == 0:
if buffer_decibel >= self.stream.THRESHOLD:
LOGGER.debug(f"[Noisegate Open] {buffer_decibel} db")
else:
LOGGER.debug(f"[Noisegate Closed] {buffer_decibel} db")
if buffer_decibel >= self.stream.THRESHOLD:
self.NG_fadeout_count = self.NG_fadeout
self.process_set_count += 1
if curr_buffer:
return bytes(curr_buffer)
else:
if self.NG_fadeout_count > 0:
self.NG_fadeout_count -= 1
LOGGER.debug(f"Frames in fadeout remaining: {self.NG_fadeout_count}")
self.process_set_count += 1
if curr_buffer:
return bytes(curr_buffer)
except OSError as e:
LOGGER.warning(e)
pass
def audio_datalist_set_volume(self, datalist, volume):
""" Change value of list of audio chunks """
sound_level = (volume / 100.)
for i in range(len(datalist)):
chunk = numpy.fromstring(datalist[i], numpy.int16)
chunk = chunk * sound_level
datalist[i] = chunk.astype(numpy.int16)
class DeviceNotFoundError(Exception):
def __init__(self):
self.devices = sounddevice.query_devices()
self.host_apis = sounddevice.query_hostapis()
super().__init__("No Devices Found")
def __str__(self):
return (
f"Devices \n"
f"{self.devices} \n "
f"Host APIs \n"
f"{pformat(self.host_apis)}"
)
def query_devices():
options = {
device.get("name"): index
for index, device in enumerate(sounddevice.query_devices())
if (device.get("max_input_channels") > 0 and device.get("hostapi") == DEFAULT)
}
if not options:
raise DeviceNotFoundError()
return options
if __name__ == '__main__':
input_index = int(input("Input:\t"))
output_index = int(input("Output:\t"))
ng = NoiseGate(_input_device_index=input_index, _output_device_index=output_index)
ng.list_devices()
ng.start()

View File

@@ -9,10 +9,11 @@ This project is intended to allow users in discord to be able to listen to their
## Setup
1. Install the pip packages found in the ```Requirements.txt``` file
2. Run ```main.py``` with Python
3. Follow the prompts in the terminal
4. Ensure your audio is playing on the selected device
5. You're all set! Ask the bot to join!
- *Optional if you are using OP25* Run `echo 0 > /sys/module/usbcore/parameters/usbfs_memory_mb`
3. Run ```main.py``` with Python
4. Follow the prompts in the terminal
5. Ensure your audio is playing on the selected device
5You're all set! Ask the bot to join!
### Understanding Audio Input
This title can be a bit confusing. The bot will display both 'input' and 'output' devices but not always *all* devices connected.
@@ -22,4 +23,9 @@ Voicemeeter is **highly** recommended for this bot. See a detailed guide on how
To change the audio source, simply delete the ```config.ini``` that was generated and restart the bot.
It will re-do the setup and allow you to select a new device.
### [To-Do](https://git.vpn.cusano.net/Discord_Bot_Gang/Discord-Radio-Bot/src/branch/master/TODO.md)
### [To-Do](https://git.vpn.cusano.net/Discord_Bot_Gang/Discord-Radio-Bot/src/branch/master/TODO.md)
**Notes for readme leter on**
- Users need to save profile after any change in discord
-

46
TODO.md
View File

@@ -1,20 +1,48 @@
## To-Do
### Main Bot
# To-Do
## Main Bot
### Main Development
#### Core
- [ ] Add new handlers for GQRX: https://github.com/gqrx-sdr/gqrx/blob/master/resources/remote-control.txt
- [ ] Add logging for Elasticstack https://www.elastic.co/guide/en/ecs-logging/python/master/installation.html
- [ ] Add a process handler to start/stop gqrx
- [ ] Fix the bug where they *disconnect* after a period of time and must be manually moved out and back in to hear them
- *May* have been fixed with the noise gate?
- [ ] Fix bug that shows different index number for audio device selection on linux
- [ ] New bug on linux shows a bunch of ALSA output
- [x] Add more proper help text with real examples for discord commands
- [ ] Add command line args with argparse for main bot
- [ ] Add method for user to change audio device without redoing entire config file
- [x] Add command line args with argparse for main bot
- [x] Add method for user to change audio device without redoing entire config file
- [ ] Need to create a method for the bot to toggle debug mode for OP25
#### Features
- [ ] Add to a linux service
- [ ] Add a function to the bot to allow *authorized* users to; update, restart, git pull, etc.
### Polishing
- [ ] Clean up code
- [ ] Transcode radio transmissions to text
### Modules
#### Willie Timer
- [ ] Revise discord help text
- [ ] Revise logging
### Final
- [ ] Create setup script
- [ ] Test setup
- [ ] Revise readme
--------------------------------------------
## Modules
### Willie Timer
- [ ] Revise the current loop, ensure it re-enters itself after triggering
- [ ] Can we use a cron somehow?
- [ ] Get more training data for WillieTimer
- [ ] Use the ```Phrases.txt``` file as the random seed?
- [ ] Figure out a way to give the model a suggestion
- [ ] Figure out a way to give the model a suggestion
### Done Previously
-------------------------------
## Dreams
- [ ] Transcode radio transmissions to text
---------------------
## Done Previously
- [x] Interact with soapysdr directly from the bot
- [x] Allow chat interaction with soapysdr
- [x] Move cogs to their own files

350
bot.py
View File

@@ -1,11 +1,11 @@
import asyncio
import logging
import os
import platform
import discord
import BotResources
import sound
import configparser
import NoiseGatev2
from discord.ext import commands
@@ -18,6 +18,9 @@ class Bot(commands.Bot):
commands.Bot.__init__(self, command_prefix=commands.when_mentioned_or(kwargs['command_prefix']),
activity=discord.Game(name=f"@ me"), status=discord.Status.idle)
# Create the logger for the bot
self.logger = logging.getLogger("Discord_Radio_Bot.Bot")
# Init the core bot variables
self.DEVICE_ID = kwargs['Device_ID']
self.DEVICE_NAME = kwargs['Device_Name']
@@ -34,13 +37,14 @@ class Bot(commands.Bot):
self.Bot_Connected = False
# Init the audio devices list
self.Devices_List = sound.query_devices().items()
self.Devices_List = NoiseGatev2.query_devices().items()
# Init radio parameters
self.profile_name = None
self.freq = "104700000"
self.mode = "wfm"
self.squelch = 0
self.profile_name = BotResources.DEFAULT_RADIO_SETTINGS['profile_name']
self.freq = BotResources.DEFAULT_RADIO_SETTINGS['freq']
self.mode = BotResources.DEFAULT_RADIO_SETTINGS['mode']
self.squelch = BotResources.DEFAULT_RADIO_SETTINGS['squelch']
self.noisegate_sensitivity = BotResources.DEFAULT_RADIO_SETTINGS['noisegate_sensitivity']
# Init SDR Variables
self.system_os_type = None
@@ -64,17 +68,23 @@ class Bot(commands.Bot):
# Add discord commands to the bot
def add_commands(self):
# Test command to see if the bot is on (help command can also be used)
@self.command(help="Use this to test if the bot is alive", brief="Sends a 'pong' in response")
async def ping(ctx):
if ctx.author.id != self.user.id:
await ctx.send('pong')
# Command to display what channel the bot is in
@self.command(help="Use this command to display what channel the bot is in",
brief="Where ya at?")
async def wya(ctx, member: discord.Member = None):
member = member or ctx.author.display_name
if self.Bot_Connected:
await ctx.send(f"Hey {str(member).capitalize()}, I'm in {ctx.voice_client.channel}")
else:
await ctx.send(f"{str(member).capitalize()}, I am not in any channel.")
# Command to join the bot the voice channel the user who called the command is in
@self.command(help="Use this command to join the bot to your channel",
brief="Joins the voice channel that the caller is in")
async def join(ctx, *, member: discord.Member = None):
member = member or ctx.author.display_name
self.logger.info(f"Join requested by {member}")
if not self.Bot_Connected:
# Wait for the bot to be ready to connect
@@ -85,52 +95,116 @@ class Bot(commands.Bot):
if discord.opus.is_loaded():
channel = ctx.author.voice.channel
self.logger.debug("Sending hello")
await ctx.send(f"Ok {str(member).capitalize()}, I'm joining {channel}")
# Join the voice channel with the audio stream
self.logger.debug('Joining')
voice_connection = await channel.connect()
# Create an audio stream from selected device
self.streamHandler = sound.PCMStream()
# Ensure the selected device is available and start the audio stream
self.streamHandler.change_device(self.DEVICE_ID)
# Play the stream
voice_connection.play(discord.PCMAudio(self.streamHandler))
self.logger.debug("Starting noisegate/stream handler")
self.streamHandler = NoiseGatev2.NoiseGate(_device_index=self.DEVICE_ID,
_voice_connection=voice_connection,
_noise_gate_threshold=self.noisegate_sensitivity)
# Start the audio stream
self.streamHandler.run()
# Start the SDR and begin playing to the audio stream
self.logger.debug("Starting SDR")
self.start_sdr()
# Change the activity to the channel and band-type being used
self.logger.debug("Changing presence")
await self.set_activity()
# 'Lock' the bot from connecting
self.logger.debug("Locking the bot")
self.Bot_Connected = True
else:
# Return that the opus library would not load
await ctx.send("Opus won't load")
self.logger.critical("OPUS didn't load properly")
else:
await ctx.send(f"{str(member).capitalize()}, I'm already connected")
self.logger.info("Bot is already in a channel")
@self.command(help="Use this command to have the bot leave your channel",
brief="Leaves the current voice channel")
async def leave(ctx):
async def leave(ctx, member: discord.Member = None):
member = member or ctx.author.display_name
self.logger.info(f"Leave requested by {member}")
if self.Bot_Connected:
# Stop the sound handlers
self.streamHandler.clean_up()
# Disconnect the client from the voice channel
await ctx.voice_client.disconnect()
self.logger.debug("Disconnecting")
await self.streamHandler.close()
self.logger.debug("Changing presence")
# Change the presence to away and '@ me'
await self.set_activity(False)
# Stop the SDR so it can cool off
self.logger.debug("Stopping SDR")
self.stop_sdr()
self.logger.debug("Unlocking the bot")
# 'Unlock' the bot
self.Bot_Connected = False
self.logger.debug("Sending Goodbye")
await ctx.send(f"Goodbye {str(member).capitalize()}.")
else:
await ctx.send(f"{str(member).capitalize()}, I'm not in a channel")
self.logger.info("Bot is not in a channel")
# Add command to change the NoiseGate threshold
@self.command(help="Use this command to change the threshold of the noisegate",
brief="Change noisegate sensitivity")
async def chngth(ctx, _threshold: int, member: discord.Member = None):
member = member or ctx.author.display_name
self.logger.info(f"Change of NoiseGate threshold requested by {member}")
await ctx.send(f"Ok {str(member).capitalize()}, I'm changing the threshold from "
f"{self.streamHandler.THRESHOLD} to {_threshold}")
self.streamHandler.THRESHOLD = _threshold
self.noisegate_sensitivity = _threshold
# Reset the profile name since we have made a change
self.profile_name = None
# If the SDR is started, restart it with the updates
await self.use_current_radio_config()
# Add commands for GQRX and OP25
if self.Handler == 'gqrx' or self.Handler == 'op25':
if self.Handler in BotResources.PDB_ACCEPTABLE_HANDLERS.keys():
# Command to display the current config
@self.command(name='displaycurprofile',
help="Use this command to display the current configuration of the bot.\n"
"Example command:\n"
"\t@ displaycurprofile",
breif="Display current bot config")
async def _displaycurprofile(ctx, member: discord.Member = None):
member = member or ctx.author.display_name
message = self.display_current_radio_config()
await ctx.send(f"Ok {str(member).capitalize()},\n{message}")
self.logger.info(f"Displaying current profile; requested by {member}")
# Command to display the current config
@self.command(name='displayprofiles',
help="Use this command to display the saved profiles.\n"
"Example command:\n"
"\t@ displayprofiles",
breif="Display current bot config")
async def _displayprofiles(ctx, member: discord.Member = None):
member = member or ctx.author.display_name
message = self.display_saved_radio_configs()
await ctx.send(f"Ok {str(member).capitalize()},\n{message}")
self.logger.info(f"Displaying all profiles; requested by {member}")
# Command to change the current frequency and mode
@self.command(name='chfreq', help="Use this command to change the frequency the bot is listening to.\n"
"Example GQRX command:\n"
"\tTune to 104.7Mhz Wideband FM (Radio) - '@ chfreq wfm 104700000\n"
@@ -140,36 +214,27 @@ class Bot(commands.Bot):
brief="Changes radio frequency")
async def chfreq(ctx, mode: str, freq: str, member: discord.Member = None):
# Possible band-types that can be used
possible_modes = []
if self.Handler == 'gqrx':
possible_modes = ['wfm', 'fm']
elif self.Handler == 'op25':
possible_modes = ['d', 'p25']
member = member or ctx.author.display_name
self.logger.info(f"{member} requested change of frequency to Mode: {mode}, Freq: {freq}")
# Check to make sure the frequency input matches the syntax needed
if len(freq) >= 6:
self.freq = freq
# Check to make sure the selected mode is valid
if mode in possible_modes:
if mode in self.possible_modes:
self.mode = mode
await ctx.send(f"Ok {str(member).capitalize()}, I'm changing the mode to "
f"{str(self.mode).upper()} and frequency to {self.freq}")
# Reset the profile name since we have made a change to the freq
# Reset the profile name since we have made a change
self.profile_name = None
# If the SDR is started, restart it with the updates
if self.sdr_started:
self.start_sdr()
await self.set_activity()
await self.use_current_radio_config()
else:
await ctx.send(f"{str(member).capitalize()}, {mode} is not valid."
f" You may only enter {possible_modes}")
f" You may only enter {self.possible_modes}")
else:
await ctx.send(f"{str(member).capitalize()}, {freq} is not valid. "
f"Please refer to the help page '@ help chfreq'")
@@ -184,13 +249,16 @@ class Bot(commands.Bot):
brief="Changes radio squelch")
async def chsquelch(ctx, squelch: float, member: discord.Member = None):
member = member or ctx.author.display_name
self.logger.info(f"{member} requested change of squelch to: {squelch}")
self.squelch = squelch
await ctx.send(f"Ok {str(member).capitalize()}, I'm changing the squelch to {self.squelch}")
# Reset the profile name since we have made a change
self.profile_name = None
# If the SDR is started, restart it with the updates
if self.sdr_started:
self.start_sdr()
await self.use_current_radio_config()
# Hidden admin commands
@self.command(name='saveprofile', hidden=True)
@@ -208,12 +276,6 @@ class Bot(commands.Bot):
else:
await ctx.send(f"{str(member).capitalize()}, there is no profile with the name '{profile_name}'")
@self.command(name='displayprofile', hidden=True)
async def _displayprofile(ctx, member: discord.Member = None):
member = member or ctx.author.display_name
message = self.display_current_radio_config()
await ctx.send(f"Ok {str(member).capitalize()}\n{message}")
# Hidden admin commands
@self.command(name='reload', hidden=True)
async def _reload(ctx, module: str, member: discord.Member = None):
@@ -225,11 +287,11 @@ class Bot(commands.Bot):
await ctx.send(f"{str(member).capitalize()}, something went wrong. Please check the console")
@self.command(name='startsdr', hidden=True)
async def _startsdr(ctx, member: discord.Member = None):
async def _startsdr(*args):
self.start_sdr()
@self.command(name='stopsdr', hidden=True)
async def _stopsdr(ctx, member: discord.Member = None):
async def _stopsdr(*args):
self.stop_sdr()
# Add discord events to the bot
@@ -239,36 +301,33 @@ class Bot(commands.Bot):
async def on_ready():
# Check the ./modules folder for any modules (cog.py)
await self.check_for_modules()
print("Bot started!")
@self.event
async def on_message(message):
await self.check_and_reply_to_ping(message)
self.logger.info("Bot started!")
# Check to see if other bots are online
async def check_other_bots_online(self):
print('Checking if other bots are online')
self.logger.info('Checking if other bots are online')
channel = self.get_channel(self.Default_Channel_ID)
print(f"Channel to be tested in: {channel}")
self.logger.debug(f"Testing in: {channel}")
bots_online = []
def verify_bot_msg(msg):
if msg.author.id in BotResources.PDB_KNOWN_BOT_IDS:
bots_online.append(msg.author.id)
if msg.author.id in BotResources.PDB_KNOWN_BOT_IDS.keys():
bots_online.append(BotResources.PDB_KNOWN_BOT_IDS[msg.author.id])
await self.wait_until_ready()
await channel.send(f"{self.Command_Prefix}ping")
# Send the ping command with the prefix the current bot is using
await channel.send(f"{self.Command_Prefix}check_modules")
seconds_waited = 0
while seconds_waited < 2:
while seconds_waited < 3:
try:
await self.wait_for("message", check=verify_bot_msg, timeout=1)
except asyncio.exceptions.TimeoutError:
seconds_waited += 1
print(f"Bots Online: {bots_online}")
self.logger.debug(f"Bots Online: {bots_online}")
if len(bots_online) == 0:
return False
@@ -278,38 +337,54 @@ class Bot(commands.Bot):
# Check the handler being used during init
def check_handler(self):
if self.Handler == "gqrx":
print("Starting GQRX handler")
self.logger.info("Starting GQRX handler")
from gqrxHandler import GQRXHandler
self.GQRXHandler = GQRXHandler()
self.GQRXHandler.start()
self.possible_modes = BotResources.PDB_ACCEPTABLE_HANDLERS['gqrx']['Modes']
elif self.Handler == 'op25':
print("Starting OP25 handler")
self.logger.info("Starting OP25 handler")
from op25Handler import OP25Handler
self.OP25Handler = OP25Handler()
self.OP25Handler.start()
self.possible_modes = BotResources.PDB_ACCEPTABLE_HANDLERS['op25']['Modes']
# Load the proper OPUS library for the device being used
def load_opus(self):
# Check the system type and load the correct library
if self.system_os_type == 'Linux_32':
discord.opus.load_opus('./opus/libopus.so')
elif self.system_os_type == 'Linux_64':
# Linux ARM AARCH64 running 32bit OS
if self.system_os_type == 'Linux_ARMv7l':
self.logger.debug(f"Loaded OPUS library for {self.system_os_type}")
discord.opus.load_opus('./opus/libopus_armv7l.so')
# Linux ARM AARCH64 running 64bit OS
if self.system_os_type == 'Linux_AARCH64':
self.logger.debug(f"Loaded OPUS library for {self.system_os_type}")
discord.opus.load_opus('./opus/libopus_aarcch64.so')
elif self.system_os_type == 'Windows':
discord.opus.load_opus('./opus/libopus.dll')
# Windows 64bit
if self.system_os_type == 'Windows_x64':
self.logger.debug(f"Loaded OPUS library for {self.system_os_type}")
discord.opus.load_opus('./opus/libopus_amd64.dll')
# Check to make sure the selected device is still available and has not changed it's index
def check_device(self):
for device, index in self.Devices_List:
if int(index) == self.DEVICE_ID and str(device) == self.DEVICE_NAME:
return True
# Check to make sure the selected device is still available and has not changed its index
def check_device(self, _override):
# Check to see if an override has been passed
if not _override:
self.logger.debug(f"Device list {self.Devices_List}")
for device, index in self.Devices_List['Input']:
if int(index) == self.DEVICE_ID and str(device) == self.DEVICE_NAME:
return True
for device, index in self.Devices_List:
if str(device) == self.DEVICE_NAME:
self.DEVICE_ID = int(index)
return True
for device, index in self.Devices_List['Input']:
if str(device) == self.DEVICE_NAME:
self.DEVICE_ID = int(index)
return True
else:
return False
else:
return False
# If an override has been passed just reply true
return True
# Search the ./modules folder for any modules to load
async def check_for_modules(self):
@@ -320,50 +395,55 @@ class Bot(commands.Bot):
if str(folder_name)[0] == '.':
continue
elif os.path.exists(os.path.join("modules", folder_name, "cog.py")):
print(f"Loaded extension: {folder_name}")
self.logger.debug(f"Loaded extension: {folder_name}")
self.load_extension(f"modules.{folder_name}.cog")
# Reload a selected module for changes
def reload_modules(self, module):
try:
self.unload_extension(f"modules.{module}.cog")
print(f"Unloaded {module}")
self.logger.debug(f"Unloaded {module}")
self.load_extension(f"modules.{module}.cog")
print(f"Loaded {module}")
self.logger.debug(f"Loaded {module}")
return True
except Exception as e:
print(e)
self.logger.error(e)
return False
# Check and store the OS type of the system for later use
def check_os_type(self):
processor = platform.machine()
if os.name == 'nt':
self.system_os_type = 'Windows'
if processor == "AMD64":
self.system_os_type = 'Windows_x64'
self.logger.debug(f"OS/Arch is {self.system_os_type}")
else:
processor = platform.architecture()[0]
if processor == "64bit":
self.system_os_type = 'Linux_64'
elif processor == "32bit":
self.system_os_type = 'Linux_32'
if processor == "aarch64":
self.system_os_type = 'Linux_AARCH64'
self.logger.debug(f"OS/Arch is {self.system_os_type}")
elif processor == "armv7l":
self.system_os_type = 'Linux_ARMv7l'
self.logger.debug(f"OS/Arch is {self.system_os_type}")
# Check to see if there is only one frequency
def start_sdr(self):
if self.Handler in ['gqrx', 'op25']:
if self.Handler in BotResources.PDB_ACCEPTABLE_HANDLERS.keys():
if type(self.freq) == str:
# Single freq sent
# Stop the SDR if it is running
self.stop_sdr()
# Start the radio
print(f"Starting freq: {self.freq}")
self.logger.debug(f"Starting freq: {self.freq}")
if self.Handler == 'gqrx':
# Set the settings in GQRX
self.GQRXHandler.set_all_settings(self.mode, self.squelch, self.freq)
self.GQRXHandler.set_gqrx_parameters(_frequency=self.freq, _squelch=self.squelch,
_fm_mode=self.mode, _output_device_name=self.DEVICE_NAME,
_start=True)
elif self.Handler == 'op25':
self.OP25Handler.set_op25_parameters(self.freq)
self.OP25Handler.start()
self.OP25Handler.set_op25_parameters(self.freq, _start=True, _output_device_name=self.DEVICE_NAME)
# Set the started variable for later checks
self.sdr_started = True
@@ -372,16 +452,22 @@ class Bot(commands.Bot):
def stop_sdr(self):
if self.sdr_started:
# Wait for the running processes to close
# Close the GQRX handler
if self.Handler == 'gqrx':
self.GQRXHandler.set_gqrx_parameters(_stop=True)
# Close the OP25 handler
if self.Handler == 'op25':
self.OP25Handler.close_op25()
self.OP25Handler.set_op25_parameters(_stop=True)
# self.OP25Handler.join()
# Need a way to 'close' GQRX
self.sdr_started = False
# Set the activity of the bot
async def set_activity(self, connected=True):
if connected:
if self.Handler in ['gqrx', 'op25']:
if self.Handler in BotResources.PDB_ACCEPTABLE_HANDLERS.keys():
if self.profile_name is None:
await self.change_presence(activity=discord.Activity(type=discord.ActivityType.listening,
name=f"{self.freq[:-1]}"
@@ -399,43 +485,64 @@ class Bot(commands.Bot):
await self.change_presence(activity=discord.Game(name=f"@ me"), status=discord.Status.idle)
# Save the current radio settings as a profile
async def save_radio_config(self, profile_name):
async def save_radio_config(self, _profile_name: str):
self.logger.debug(f"Saving profile {_profile_name}")
config = configparser.SafeConfigParser()
if os.path.exists('./profiles.ini'):
config.read('./profiles.ini')
profile_name = str(_profile_name).upper()
if not config.has_section(str(profile_name)):
config.add_section(str(profile_name))
config[str(profile_name)]['Frequency'] = self.freq
config[str(profile_name)]['Mode'] = self.mode
config[str(profile_name)]['Squelch'] = str(self.squelch)
config[str(profile_name)]['Noisegate_Sensitivity'] = str(self.noisegate_sensitivity)
with open('./profiles.ini', 'w+') as config_file:
config.write(config_file)
self.profile_name = profile_name
await self.use_current_radio_config()
async def use_current_radio_config(self):
if self.sdr_started:
self.start_sdr()
# Set the loaded profile settings into GQRX
if self.Handler == "gqrx":
self.GQRXHandler.set_gqrx_parameters(_frequency=self.freq, _squelch=self.squelch,
_fm_mode=self.mode)
# Restart OP25 to use the loaded profile
if self.Handler == "op25":
self.start_sdr()
# Set the activity to reflect the loaded profile
await self.set_activity()
# Load a saved profile into the current settings
async def load_radio_config(self, profile_name):
self.logger.debug(f"Loading profile {profile_name}")
config = configparser.ConfigParser()
if os.path.exists('./profiles.ini'):
config.read('./profiles.ini')
if config.has_section(str(profile_name)):
self.profile_name = profile_name
self.freq = config[str(profile_name)]['Frequency']
self.mode = config[str(profile_name)]['Mode']
self.squelch = float(config[str(profile_name)]['Squelch'])
if self.sdr_started:
self.start_sdr()
await self.set_activity()
if config.has_section(str(profile_name).upper()):
self.profile_name = str(profile_name).upper()
self.freq = config[self.profile_name]['Frequency']
self.mode = config[self.profile_name]['Mode']
self.squelch = float(config[self.profile_name]['Squelch'])
try:
self.noisegate_sensitivity = int(config[self.profile_name]['Noisegate_Sensitivity'])
except KeyError:
self.logger.warning(f"Config does not contain a 'noisegate sensitivity' value, "
f"creating one now with the default value: "
f"{BotResources.DEFAULT_NOISEGATE_THRESHOLD}")
self.noisegate_sensitivity = BotResources.DEFAULT_NOISEGATE_THRESHOLD
await self.save_radio_config(self.profile_name)
await self.use_current_radio_config()
return True
else:
return False
@@ -445,20 +552,29 @@ class Bot(commands.Bot):
def display_current_radio_config(self):
message_body = ""
if self.profile_name:
message_body += f"Profile Name: {self.profile_name}\n"
message_body += f"Frequency: {self.freq}\n" \
f"Mode: {self.mode}\n"
message_body += f"Profile Name: {str(self.profile_name).upper()}\n"
message_body += f"\tMode:\t\t\t\t\t{self.mode}\n" \
f"\tFrequency:\t\t\t{self.freq}\n" \
f"\tNoisegate Sensitivity:\t{self.noisegate_sensitivity}"
if self.squelch:
message_body += f"Squelch: {self.squelch}"
message_body += f"\tSquelch:\t\t\t\t{self.squelch}"
return message_body
# Check if message is a ping request and respond even if it is a bot
async def check_and_reply_to_ping(self, message):
if "ping" in message.content:
ctx = await self.get_context(message)
await self.invoke(ctx)
return True
else:
await self.process_commands(message)
return False
def display_saved_radio_configs(self):
message_body = f"Saved configs\n"
config = configparser.ConfigParser()
if os.path.exists('./profiles.ini'):
config.read('./profiles.ini')
for section in config.sections():
message_body += f"\nProfile Name: {section}:\n" \
f"\tMode:\t\t\t\t\t{config[section]['Mode']}\n" \
f"\tFrequency:\t\t\t{config[section]['Frequency']}\n"
try:
message_body += f"\tNoisegate Sensitivity:\t{config[section]['Noisegate_Sensitivity']}\n"
except KeyError:
self.logger.warning(f"Config does not contain a 'noisegate sensitivity' value. Please load the profile")
message_body += f"\tSquelch:\t\t\t\t{config[section]['Squelch']}\n"
return message_body

View File

@@ -1,48 +1,239 @@
import configparser
import shutil
import logging
import threading
import subprocess
import time
from pathlib import Path
from telnetlib import Telnet
from BotResources import check_negative
from BotResources import *
from time import sleep
class GQRXHandler():
def __init__(self, hostname: str = "localhost", port: int = 7356):
self.hostname = hostname
self.port = port
self.telnet_connection = None
def reset_crashed(_config_path):
config = configparser.SafeConfigParser()
config.read(_config_path)
if config.has_section('General'):
if config.getboolean('General', 'crashed'):
config['General']['crashed'] = 'false'
with open(_config_path, 'w') as config_file:
config.write(config_file)
return True
else:
return False
def enable_agc(_config_path):
config = configparser.SafeConfigParser()
config.read(_config_path)
if config.has_option('receiver', 'agc_off'):
config.remove_option('receiver', 'agc_off')
with open(_config_path, 'w') as config_file:
config.write(config_file)
return True
class GQRXHandler(threading.Thread):
def __init__(self):
super().__init__()
self.GQRX_Config_Path = Path(f"{Path.home()}/.config/gqrx/drb_defaults.conf")
self.GQRXDir: str = GQRX_BIN_LOCATION
self.GQRXEXE: str = shutil.which(GQRX_BIN)
self.GQRXProc = None
self.GQRX_Started = False
self.logger = logging.getLogger("Discord_Radio_Bot.GQRXHandler")
self.Frequency = None
self.Mode = DEFAULT_RADIO_SETTINGS['mode']
self.Frequency = DEFAULT_RADIO_SETTINGS['freq']
self.Squelch = DEFAULT_RADIO_SETTINGS['squelch']
self.Start_GQRX = False
self.Stop_GQRX = False
self.Output_Device_Name = None
self.hostname = "localhost"
self.port = 7356
self.tel_conn = None
def run(self) -> None:
while True:
if self.Start_GQRX:
self.open_gqrx()
self.Start_GQRX = False
self.Stop_GQRX = False
self.logger.debug("GQRX is open, waiting for it to close")
while not self.Stop_GQRX:
sleep(1)
self.logger.debug('Request to close GQRX')
self.close_gqrx()
sleep(.5)
def set_gqrx_parameters(self, _frequency: str = False, _start: bool = False, _stop: bool = False,
_output_device_name: str = None, _fm_mode: str = None, _squelch: float = None,
_hostname: str = None, _port: int = None, _start_dsp: bool = None):
if _frequency:
self.Frequency = _frequency
if self.GQRX_Started:
self.change_freq(_frequency)
if _output_device_name:
self.Output_Device_Name = _output_device_name
if _fm_mode:
self.Mode = _fm_mode
if self.GQRX_Started:
self.change_mode(_fm_mode)
if _squelch:
self.Squelch = _squelch
if self.GQRX_Started:
self.change_squelch(_squelch)
if _hostname:
self.hostname = _hostname
self.Start_GQRX = True
if _port:
self.port = _port
self.Start_GQRX = True
if _start_dsp:
self.start_dsp()
if _start:
self.Start_GQRX = _start
if _stop:
self.Stop_GQRX = _stop
def open_gqrx(self):
if self.GQRX_Started:
self.close_gqrx()
gqrx_kwargs = [f"gqrx", "-c", "drb_defaults.conf"]
self.logger.info(f"Resetting 'crashed' option in the GQRX config")
self.reset_or_create_config()
self.logger.info(f"Starting GQRX")
self.logger.debug(f"GQRX Keyword Args: {gqrx_kwargs}")
self.GQRXProc = subprocess.Popen(gqrx_kwargs, executable=self.GQRXEXE, shell=False)
while not self.tel_conn:
self.create_telnet_connection()
sleep(2)
self.logger.debug(f"Waiting for GQRX to start")
self.GQRX_Started = True
self.start_dsp()
self.set_all_settings(_squelch=self.Squelch, _mode=self.Mode, _freq=self.Frequency)
self.logger.debug('Finished opening GQRX')
def close_gqrx(self):
self.logger.info(f"Closing GQRX")
try:
self.GQRXProc.kill()
seconds_waited = 0
while self.GQRXProc.poll() is None:
# Terminate the process every 5 seconds
if seconds_waited % 5 == 0:
self.logger.info("Terminating GQRX")
self.GQRXProc.terminate()
sleep(1)
self.logger.debug(f"Waited {seconds_waited} seconds")
seconds_waited += 1
self.logger.debug("GQRX Closed")
self.GQRX_Started = False
self.tel_conn = None
except Exception as e:
self.logger.error(e)
def create_telnet_connection(self):
print("Creating connection")
tel_conn = Telnet(self.hostname, self.port)
tel_conn.open(self.hostname, self.port)
self.logger.debug("Creating connection")
try:
self.tel_conn = Telnet(self.hostname, self.port)
self.tel_conn.open(self.hostname, self.port)
self.logger.debug(f"GQRX is open")
return True
except Exception as err:
self.logger.warning(err)
self.tel_conn = None
return False
return tel_conn
def check_dsp(self):
self.logger.debug(f"Checking if DSP is running on GQRX")
self.tel_conn.write(bytes(f"u DSP", 'utf-8'))
if self.tel_conn.read_some() == b"1":
return True
else:
return False
def start_dsp(self):
if not self.check_dsp():
self.logger.debug(f"Starting DSP on GQRX")
self.tel_conn.write(bytes(f"U DSP 1", 'utf-8'))
self.tel_conn.read_until(b'RPRT 0')
def change_freq(self, freq):
tel_conn = self.create_telnet_connection()
print(f"Changing freq to {freq}")
tel_conn.write(bytes(f"F {int(freq)}", 'utf-8'))
sleep(1)
tel_conn.close()
self.logger.debug(f"Changing freq to {freq}")
self.tel_conn.write(bytes(f"F {int(freq)}", 'utf-8'))
self.tel_conn.read_until(b'RPRT 0')
def change_squelch(self, squelch):
tel_conn = self.create_telnet_connection()
if not check_negative(squelch):
squelch = float(-abs(squelch))
print(f"Changing squelch to {squelch}")
tel_conn.write(bytes(f"L SQL {float(squelch)}", 'utf-8'))
sleep(1)
tel_conn.close()
self.logger.debug(f"Changing squelch to {squelch}")
self.tel_conn.write(bytes(f"L SQL {float(squelch)}", 'utf-8'))
self.tel_conn.read_until(b'RPRT 0')
def change_mode(self, mode):
tel_conn = self.create_telnet_connection()
print(f"Changing mode to {mode}")
tel_conn.write(bytes(f"M {str(mode)}", 'utf-8'))
sleep(1)
self.logger.debug(f"Changing mode to {mode}")
self.tel_conn.write(bytes(f"M {str(mode)}", 'utf-8'))
self.tel_conn.read_until(b'RPRT 0')
tel_conn.close()
def set_all_settings(self, _mode, _squelch, _freq):
self.change_squelch(0)
self.change_mode(_mode)
self.change_freq(_freq)
self.change_squelch(_squelch)
def set_all_settings(self, mode, squelch, freq):
self.change_mode(mode)
self.change_freq(freq)
self.change_squelch(squelch)
def creat_config(self):
from templates.gqrx_config_template import drb_defaults
config = drb_defaults
try:
with open(self.GQRX_Config_Path, 'w+') as config_file:
config_file.write(config)
except OSError as err:
self.logger.error(err)
def reset_or_create_config(self):
if self.GQRX_Config_Path.is_file():
try:
self.logger.debug(f"Enabling AGC in the GQRX config")
enable_agc(_config_path=self.GQRX_Config_Path)
self.logger.debug(f"GQRX Config exists, resetting 'crashed' setting")
reset_crashed(_config_path=self.GQRX_Config_Path)
except configparser.DuplicateOptionError as err:
self.logger.warning(err)
self.creat_config()
else:
self.logger.debug(f"GQRX config does not exist, creating it from template")
self.creat_config()

111
main.py
View File

@@ -1,8 +1,9 @@
import logging
import os
import time
import bot
import argparse
from BotResources import check_if_config_exists, write_config_file, read_config_file
import BotResources
# Jorn
#token = 'OTE1MDY0OTk2OTk0NjMzNzI5.YaWKsA.Y9yaCGg_VXRL_qQVbs05vo7gSAc'
@@ -15,52 +16,120 @@ from BotResources import check_if_config_exists, write_config_file, read_config_
#name = "VoiceMeeter Output"
LOGGER = logging.getLogger("Discord_Radio_Bot.Main")
class BotDeviceNotFound(Exception):
def __init__(self, device):
print(f"Unable to find the device: {device}")
LOGGER.debug(f"Unable to find the device: {device}")
try:
os.remove('./config.ini')
except OSError:
print("Config file not found, restarting.")
LOGGER.warning("Config file not found, restarting.")
#os.execv(__file__, sys.argv)
def main(**passed_config):
print('Checking config file...')
if not check_if_config_exists():
print("No config file exists, please enter this information now")
write_config_file(init=True, **passed_config)
# Don't create a config if the user passed parameters
if len(passed_config.keys()) == 0:
LOGGER.info('Checking config file...')
if not BotResources.check_if_config_exists():
LOGGER.warning("No config file exists, please enter this information now")
BotResources.write_config_file(init=True)
config = read_config_file()
config = BotResources.read_config_file()
# Overwrite config options if they were passed
for sub in config:
# checking if key present in other dictionary
if sub in passed_config:
config[sub] = passed_config[sub]
if len(passed_config.keys()) == 0:
for sub in config:
# checking if key present in other dictionary
if sub in passed_config and passed_config[sub]:
config[sub] = passed_config[sub]
print('Starting Bot...')
LOGGER.info('Starting Bot...')
discord_bot_client = bot.Bot(Token=config['Bot Token'], Device_ID=config['Device ID'], Device_Name=config['Device Name'],
Mention_Group=config['Mention Group'], Channel_ID=config['Channel ID'], Handler=config['Handler'])
discord_bot_client = bot.Bot(Token=config['Bot Token'], Device_ID=config['Device ID'],
Device_Name=config['Device Name'], Mention_Group=config['Mention Group'],
Channel_ID=config['Channel ID'], Handler=config['Handler'])
print(f"Verifying audio device:\t{config['Device Name']}")
LOGGER.debug(f"Verifying audio device:\t{config['Device Name']}")
if not discord_bot_client.check_device():
if not discord_bot_client.check_device(_override=bool("Device ID" in passed_config.keys())):
raise BotDeviceNotFound(config['Device Name'])
discord_bot_client.start_bot()
def cmd_arguments():
parser = argparse.ArgumentParser(description='Discord Bot Gang - Premium Discord Bot\n'
'The more you listen, the less you can see')
# Add the arguments
# Arg to override bot token
parser.add_argument('-t', '--token',
metavar='Bot Token',
dest='Bot Token',
type=str,
help='Override saved bot token')
# Arg to override the input device in the config
parser.add_argument('-i', '--input_device_id',
metavar='Device ID',
dest='Device ID',
type=int,
help='Override saved input device')
# Arg to override mention group
parser.add_argument('-m', '--mention',
metavar='Mention Group',
dest='Mention Group',
type=str,
help='Override saved mention group')
# Arg to override default channel to send messages
parser.add_argument('-c', '--channel',
metavar='Channel ID',
dest='Channel ID',
type=int,
help='Override saved sending channel')
# Arg to override handler
parser.add_argument('-u', '--handler',
metavar='Handler',
dest='Handler',
type=str,
help='Override saved SDR handler')
# Arg to save the overridden arguments
#parser.add_argument('-s', '--save',
# metavar='save_overrides',
# dest='save_overrides',
# type=str,
# help='Save the overridden arguments passed')
# Retrieve the args
args = vars(parser.parse_args())
return args
if __name__ == '__main__':
cmd_args = cmd_arguments()
BotResources.init_global_logger()
if not all(cmd_args.values()):
LOGGER.debug("Passed arguments:")
for arg in cmd_args:
if cmd_args[arg]:
LOGGER.debug(f"\t{arg}:\t{cmd_args[arg]}")
try:
print('Starting...')
LOGGER.info('Starting...')
while True:
try:
main()
main(**cmd_args)
except BotDeviceNotFound:
print("Restarting...")
LOGGER.info("Restarting...")
time.sleep(2)
except KeyboardInterrupt:
print("Exiting...")
LOGGER.error("Exiting...")

View File

@@ -0,0 +1,52 @@
import logging
from discord.ext import commands
LOGGER = logging.getLogger("Discord_Radio_Bot.LinkCop")
class ClearMessages(commands.Cog):
def __init__(self, bot):
self.Bot = bot
@commands.command(name='clear',
help="Use this command to clear a given number of messages from the channel it is called in.\n"
"There is a limit of 100 messages. Please be patient, it may take a while to process a large request."
"Example command:\n"
"\t@ clear\n"
"\t@ clear 10",
breif="Clear x messages in the channel it's called in")
async def clear(self, ctx, amount=2):
member = ctx.author.display_name
member_id = ctx.author.id
mtn_member = f"<@{member_id}>"
if isinstance(amount, int):
if not amount > 0:
ctx.channel.send(f"{member}, the number needs to be positive...")
else:
LOGGER.info(f"Clear {amount} messages requested by {member}")
authors = {}
async for message in ctx.channel.history(limit=amount):
if message.author not in authors:
authors[message.author] = 1
else:
authors[message.author] += 1
await message.delete()
msg = f"{mtn_member}, I deleted {sum(authors.values())} messages from {len(authors.keys())} users:\n"
LOGGER.debug(f"Deleted {sum(authors.values())} messages from {ctx.message.channel}")
for author in authors.keys():
msg += f"\t{str(author).split('#', 1)[0]}: {authors[author]}\n"
LOGGER.debug(f"Deleted {authors[author]} messages from {author}")
await ctx.channel.send(msg)
else:
ctx.channel.send(f"{member}, you should check out the 'help' section...")
def setup(bot: commands.Bot):
bot.add_cog(ClearMessages(bot))

View File

@@ -1,8 +1,9 @@
import logging
import re
import discord
from discord.ext import commands
import random
from BotResources import PDB_KNOWN_BOT_IDS
from BotResources import PDB_KNOWN_BOT_IDS, check_and_reply_to_ping
regex_link = re.compile('(http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\(\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+)')
@@ -26,6 +27,8 @@ random_message = ["{%mtn_user%}, tsk tsk. Links belong here:\n{%ref_og_msg%}", "
# welcome 757379843792044102
# the-round-table 367396189529833474
LOGGER = logging.getLogger("Discord_Radio_Bot.LinkCop")
class LinkCop(commands.Cog):
def __init__(self, bot):
self.bot = bot
@@ -44,9 +47,7 @@ class LinkCop(commands.Cog):
# Bring in the known bot IDs from PDB bots
self.whitelisted_ID = PDB_KNOWN_BOT_IDS
self.whitelisted_ID.append([
235148962103951360 # Carl Bot
])
self.whitelisted_ID['Carl Bot'] = 235148962103951360
self.add_events()
@@ -61,7 +62,7 @@ class LinkCop(commands.Cog):
try:
if not any(item.id in self.whitelisted_groups for item in ctx.author.roles):
if check_message(ctx.content):
print('Msg with links detected in a blocked channel')
LOGGER.info('Msg with links detected in a blocked channel')
response_text = self.generate_response(ctx)
@@ -70,10 +71,10 @@ class LinkCop(commands.Cog):
# Delete the original message
await ctx.delete()
except AttributeError as err:
print(f"Link Cop Error: '{err}'")
print(f"Bot or other non-user that has not been whitelisted sent a message")
LOGGER.error(f"Link Cop Error: '{err}'\nBot or other non-user that has not "
f"been whitelisted sent a message")
await self.bot.check_and_reply_to_ping(ctx)
await check_and_reply_to_ping(self.bot, ctx)
async def send_message(self, message):
send_channel = self.bot.get_channel(id=self.reply_channel_id)
@@ -125,7 +126,7 @@ def check_message(message_text):
def get_links(message_text):
links = regex_link.findall(message_text)
print(links)
LOGGER.error(links)
if len(links) > 0:
return links

View File

@@ -0,0 +1,24 @@
from discord.ext import commands
from BotResources import check_and_reply_to_ping
class ModuleProbe(commands.Cog):
def __init__(self, bot):
self.bot = bot
self.channel_id = 767303243285790721
self.add_events()
def add_events(self):
@self.bot.event
async def on_message(message):
await check_and_reply_to_ping(self.bot, message)
@commands.command(help="This command is used by other bots to test if there are any bots online with modules.")
async def check_modules(self, ctx):
if ctx.author.id != self.bot.user.id:
await ctx.send('pong')
def setup(bot: commands.Bot):
bot.add_cog(ModuleProbe(bot))

View File

@@ -1,44 +1,81 @@
import threading
import logging
import shutil
import subprocess
import asyncio
import os
import threading
import time
from BotResources import DEFAULT_RADIO_SETTINGS
class OP25Handler: #(threading.Thread):
class OP25Handler(threading.Thread):
def __init__(self):
super().__init__()
self.OP25Dir: str = "/home/pi/op25/op25/gr-op25_repeater/apps"
self.OP25EXE: str = shutil.which("/home/pi/op25/op25/gr-op25_repeater/apps/rx.py")
self.OP25Proc = None
self.Frequency = None
self.Frequency = DEFAULT_RADIO_SETTINGS['freq']
def start(self) -> None:
self.open_op25()
self.HTTP_ENABLED = False
def set_op25_parameters(self, _frequency):
self.Frequency = _frequency
self.Start_OP25 = False
self.Stop_OP25 = False
self.Output_Device_Name = None
self.logger = logging.getLogger("Discord_Radio_Bot.OP25Handler")
def run(self) -> None:
while True:
if self.Start_OP25:
self.open_op25()
self.Start_OP25 = False
self.Stop_OP25 = False
while not self.Stop_OP25:
time.sleep(1)
self.close_op25()
time.sleep(.5)
def set_op25_parameters(self, _frequency: str = False, _http_enabled: bool = True, _start: bool = False,
_stop: bool = False, _output_device_name: str = None):
if _frequency:
self.Frequency = _frequency
if _start:
self.Start_OP25 = _start
if _stop:
self.Stop_OP25 = _stop
if _http_enabled:
self.HTTP_ENABLED = _http_enabled
if _output_device_name:
self.Output_Device_Name = _output_device_name
def open_op25(self):
if self.OP25Proc is not None:
self.close_op25()
print(f"Starting OP25")
p25_kwargs = [f"./rx.py", "--args", "rtl", "-N", "LNA:49", "-s", "200000", "-o", "25600", "-w", "-U", "-O",
f"{self.Output_Device_Name}", "-f", f"{self.Frequency}e6", "-X", "-2"]
cwd = os.getcwd()
print(cwd)
self.logger.info(f"Starting OP25")
# Change the interpreter's working directory (idr why)
if self.HTTP_ENABLED:
p25_kwargs.extend(["-l", "http:0.0.0.0:8080"])
os.chdir(self.OP25Dir)
self.logger.debug(f"OP25 Keyword Args: {p25_kwargs}")
print(os.getcwd())
self.OP25Proc = subprocess.Popen([f"./rx.py", "--args", "rtl", "-N", "LNA:49", "-s",
"200000", "-o", "25600", "-U", "-f", f"{self.Frequency}e6", "-X", "-2",
"-l" "http:0.0.0.0:8080"])
os.chdir(cwd)
self.OP25Proc = subprocess.Popen(p25_kwargs, executable=self.OP25EXE, shell=False, cwd=self.OP25Dir,
stdin=subprocess.DEVNULL, stdout=subprocess.DEVNULL, stderr=subprocess.STDOUT)
def close_op25(self):
print(f"Closing OP25")
self.logger.info(f"Closing OP25")
try:
self.OP25Proc.kill()
@@ -46,11 +83,11 @@ class OP25Handler: #(threading.Thread):
while self.OP25Proc.poll() is None:
# Terminate the process every 5 seconds
if seconds_waited % 5 == 0:
print("Terminating OP25")
self.logger.info("Terminating OP25")
self.OP25Proc.terminate()
asyncio.sleep(1000)
print(f"Waited {seconds_waited} seconds")
time.sleep(1)
self.logger.debug(f"Waited {seconds_waited} seconds")
seconds_waited += 1
except Exception as e:
print(e)
self.logger.error(e)

View File

@@ -1,3 +1,6 @@
discord~=1.7.3
sounddevice~=0.4.3
pynacl~=1.5.0
numpy==1.22.3
scipy==1.8.0
matplotlib~=3.5.1
pyrtlsdr~=0.2.92
PyAudio~=0.2.11

View File

@@ -1,62 +0,0 @@
import sounddevice as sd
from pprint import pformat
DEFAULT = 0
sd.default.channels = 2
sd.default.dtype = "int16"
sd.default.latency = "low"
sd.default.samplerate = 48000
class PCMStream:
globals()
def __init__(self):
self.stream = None
def read(self, num_bytes):
# frame is 4 bytes
frames = int(num_bytes / 4)
data = self.stream.read(frames)[0]
# convert to pcm format
return bytes(data)
def change_device(self, device_ID):
self.clean_up()
self.stream = sd.RawInputStream(device=device_ID)
self.stream.start()
def clean_up(self):
if self.stream is not None:
self.stream.stop()
self.stream.close()
class DeviceNotFoundError(Exception):
def __init__(self):
self.devices = sd.query_devices()
self.host_apis = sd.query_hostapis()
super().__init__("No Devices Found")
def __str__(self):
return (
f"Devices \n"
f"{self.devices} \n "
f"Host APIs \n"
f"{pformat(self.host_apis)}"
)
def query_devices():
options = {
device.get("name"): index
for index, device in enumerate(sd.query_devices())
if (device.get("max_input_channels") > 0 and device.get("hostapi") == DEFAULT)
}
if not options:
raise DeviceNotFoundError()
return options

0
templates/__init__.py Normal file
View File

View File

@@ -0,0 +1,41 @@
drb_defaults = """[General]
configversion=2
crashed=false
[audio]
gain=-20
udp_host=localhost
[dxcluster]
DXCAddress=localhost
DXCFilter=
DXCPort=7300
DXCSpotTimeout=10
DXCUsername=nocall
[fft]
averaging=85
fft_window=5
waterfall_span=1
[gui]
geometry=@ByteArray(\x1\xd9\xd0\xcb\0\x3\0\0\0\0\x4t\0\0\0$\0\0\a\x7f\0\0\x3\x12\0\0\x4v\0\0\0\x42\0\0\a\x7f\0\0\x3\x12\0\0\0\0\0\0\0\0\a\x80\0\0\x4v\0\0\0\x42\0\0\a\x7f\0\0\x3\x12)
state=@ByteArray(\0\0\0\xff\0\0\0\0\xfd\0\0\0\x2\0\0\0\x1\0\0\x1G\0\0\x2t\xfc\x2\0\0\0\x2\xfc\0\0\0\x42\0\0\x1\x89\0\0\x1\x89\0\b\0!\xfa\0\0\0\x1\x2\0\0\0\x3\xfb\0\0\0\x18\0\x44\0o\0\x63\0k\0I\0n\0p\0u\0t\0\x43\0t\0l\x1\0\0\0\0\xff\xff\xff\xff\0\0\x1P\0\xff\xff\xff\xfb\0\0\0\x12\0\x44\0o\0\x63\0k\0R\0x\0O\0p\0t\x1\0\0\0\0\xff\xff\xff\xff\0\0\x1g\0\a\xff\xff\xfb\0\0\0\xe\0\x44\0o\0\x63\0k\0\x46\0\x66\0t\x1\0\0\0\0\xff\xff\xff\xff\0\0\0\xc8\0\a\xff\xff\xfc\0\0\x1\xd1\0\0\0\xe5\0\0\0\xc3\0\xff\xff\xff\xfa\0\0\0\0\x2\0\0\0\x2\xfb\0\0\0\x12\0\x44\0o\0\x63\0k\0\x41\0u\0\x64\0i\0o\x1\0\0\0\0\xff\xff\xff\xff\0\0\0\xc3\0\xff\xff\xff\xfb\0\0\0\xe\0\x44\0o\0\x63\0k\0R\0\x44\0S\0\0\0\0\0\xff\xff\xff\xff\0\0\0h\0\xff\xff\xff\0\0\0\x3\0\0\0\0\0\0\0\0\xfc\x1\0\0\0\x1\xfb\0\0\0\x1a\0\x44\0o\0\x63\0k\0\x42\0o\0o\0k\0m\0\x61\0r\0k\0s\0\0\0\0\0\xff\xff\xff\xff\0\0\x1\x42\0\xff\xff\xff\0\0\x1\xbd\0\0\x2t\0\0\0\x1\0\0\0\x2\0\0\0\b\0\0\0\x2\xfc\0\0\0\x1\0\0\0\x2\0\0\0\x1\0\0\0\x16\0m\0\x61\0i\0n\0T\0o\0o\0l\0\x42\0\x61\0r\x1\0\0\0\0\xff\xff\xff\xff\0\0\0\0\0\0\0\0)
[input]
dc_cancel=true
device="rtl=0"
frequency=154785000
gains=@Variant(\0\0\0\b\0\0\0\x1\0\0\0\x6\0L\0N\0\x41\0\0\0\x2\0\0\x1P)
sample_rate=1800000
[receiver]
demod=3
filter_high_cut=2500
filter_low_cut=-2500
offset=-590400
sql_level=-50
[remote_control]
allowed_hosts=localhost, 127.0.0.1, ::1
enabled=true"""