Skip to content

Commit b1c8d35

Browse files
committed
Moved shared.py functions to respective files, updated function call references & isort imports
1 parent fcfcc0a commit b1c8d35

File tree

11 files changed

+306
-289
lines changed

11 files changed

+306
-289
lines changed

src/api.py

Lines changed: 14 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -57,21 +57,23 @@
5757
"""
5858

5959
import base64
60-
import ConfigParser
6160
import errno
6261
import hashlib
63-
import httplib
6462
import json
6563
import random # nosec
6664
import socket
6765
import subprocess
6866
import time
69-
import xmlrpclib
7067
from binascii import hexlify, unhexlify
71-
from SimpleXMLRPCServer import SimpleXMLRPCRequestHandler, SimpleXMLRPCServer
7268
from struct import pack
7369

70+
import ConfigParser
71+
import httplib
72+
import xmlrpclib
73+
from SimpleXMLRPCServer import SimpleXMLRPCRequestHandler, SimpleXMLRPCServer
74+
7475
import defaults
76+
import helper_addressbook
7577
import helper_inbox
7678
import helper_sent
7779
import network.stats
@@ -80,18 +82,14 @@
8082
import shared
8183
import shutdown
8284
import state
83-
from addresses import (
84-
addBMIfNotPresent,
85-
calculateInventoryHash,
86-
decodeAddress,
87-
decodeVarint,
88-
varintDecodeError
89-
)
85+
from addresses import (addBMIfNotPresent, calculateInventoryHash,
86+
decodeAddress, decodeVarint, varintDecodeError)
9087
from bmconfigparser import BMConfigParser
9188
from debug import logger
9289
from helper_sql import SqlBulkExecute, sqlExecute, sqlQuery, sqlStoredProcedure
9390
from inventory import Inventory
9491
from network.threads import StoppableThread
92+
from storage import filesystem
9593
from version import softwareVersion
9694

9795
try: # TODO: write tests for XML vulnerabilities
@@ -214,8 +212,8 @@ def run(self):
214212
if BMConfigParser().safeGet(
215213
'bitmessagesettings', 'apivariant') == 'json':
216214
try:
217-
from jsonrpclib.SimpleJSONRPCServer import (
218-
SimpleJSONRPCServer as RPCServerBase)
215+
from jsonrpclib.SimpleJSONRPCServer import \
216+
SimpleJSONRPCServer as RPCServerBase
219217
except ImportError:
220218
logger.warning(
221219
'jsonrpclib not available, failing back to XML-RPC')
@@ -863,7 +861,7 @@ def HandleDeleteAddress(self, address):
863861
13, 'Could not find this address in your keys.dat file.')
864862
self.config.save()
865863
queues.UISignalQueue.put(('writeNewAddressToTable', ('', '', '')))
866-
shared.reloadMyAddressHashes()
864+
filesystem.reloadMyAddressHashes()
867865
return "success"
868866

869867
@command('getAllInboxMessages')
@@ -1205,7 +1203,7 @@ def HandleAddSubscription(self, address, label=''):
12051203
raise APIError(16, 'You are already subscribed to that address.')
12061204
sqlExecute(
12071205
"INSERT INTO subscriptions VALUES (?,?,?)", label, address, True)
1208-
shared.reloadBroadcastSendersForWhichImWatching()
1206+
helper_addressbook.reloadBroadcastSendersForWhichImWatching()
12091207
queues.UISignalQueue.put(('rerenderMessagelistFromLabels', ''))
12101208
queues.UISignalQueue.put(('rerenderSubscriptions', ''))
12111209
return 'Added subscription.'
@@ -1219,7 +1217,7 @@ def HandleDeleteSubscription(self, address):
12191217

12201218
address = addBMIfNotPresent(address)
12211219
sqlExecute("DELETE FROM subscriptions WHERE address=?", address)
1222-
shared.reloadBroadcastSendersForWhichImWatching()
1220+
helper_addressbook.reloadBroadcastSendersForWhichImWatching()
12231221
queues.UISignalQueue.put(('rerenderMessagelistFromLabels', ''))
12241222
queues.UISignalQueue.put(('rerenderSubscriptions', ''))
12251223
return 'Deleted subscription if it existed.'

src/bitmessagecurses/__init__.py

Lines changed: 12 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -10,27 +10,29 @@
1010
# * python2-pythondialog
1111
# * dialog
1212

13-
import ConfigParser
1413
import curses
1514
import os
1615
import sys
1716
import time
1817
from textwrap import fill
1918
from threading import Timer
2019

20+
import ConfigParser
2121
from dialog import Dialog
22+
23+
import helper_addressbook
2224
import helper_sent
2325
import l10n
2426
import network.stats
2527
import queues
2628
import shared
2729
import shutdown
2830
import state
29-
3031
from addresses import addBMIfNotPresent, decodeAddress
3132
from bmconfigparser import BMConfigParser
3233
from helper_sql import sqlExecute, sqlQuery
3334
from inventory import Inventory
35+
from storage import filesystem
3436

3537
# pylint: disable=global-statement
3638

@@ -635,15 +637,15 @@ def handlech(c, stdscr):
635637
else:
636638
addresses[addrcur][3] = 0 # black
637639
addresses[addrcur][1] = True
638-
shared.reloadMyAddressHashes() # Reload address hashes
640+
filesystem.reloadMyAddressHashes() # Reload address hashes
639641
elif t == "5": # Disable address
640642
a = addresses[addrcur][2]
641643
BMConfigParser().set(a, "enabled", "false") # Set config
642644
addresses[addrcur][3] = 8 # Set color to gray
643645
# Write config
644646
BMConfigParser().save()
645647
addresses[addrcur][1] = False
646-
shared.reloadMyAddressHashes() # Reload address hashes
648+
filesystem.reloadMyAddressHashes() # Reload address hashes
647649
elif t == "6": # Delete address
648650
r, t = d.inputbox("Type in \"I want to delete this address\"", width=50)
649651
if r == d.DIALOG_OK and t == "I want to delete this address":
@@ -703,7 +705,7 @@ def handlech(c, stdscr):
703705
r, t = d.inputbox("New subscription address")
704706
if r == d.DIALOG_OK:
705707
addr = addBMIfNotPresent(t)
706-
if not shared.isAddressInMySubscriptionsList(addr):
708+
if not helper_addressbook.isAddressInMySubscriptionsList(addr):
707709
r, t = d.inputbox("New subscription label")
708710
if r == d.DIALOG_OK:
709711
label = t
@@ -713,29 +715,29 @@ def handlech(c, stdscr):
713715
subscriptions.reverse()
714716

715717
sqlExecute("INSERT INTO subscriptions VALUES (?,?,?)", label, addr, True)
716-
shared.reloadBroadcastSendersForWhichImWatching()
718+
helper_addressbook.reloadBroadcastSendersForWhichImWatching()
717719
elif t == "2":
718720
r, t = d.inputbox("Type in \"I want to delete this subscription\"")
719721
if r == d.DIALOG_OK and t == "I want to delete this subscription":
720722
sqlExecute(
721723
"DELETE FROM subscriptions WHERE label=? AND address=?",
722724
subscriptions[subcur][0],
723725
subscriptions[subcur][1])
724-
shared.reloadBroadcastSendersForWhichImWatching()
726+
helper_addressbook.reloadBroadcastSendersForWhichImWatching()
725727
del subscriptions[subcur]
726728
elif t == "3":
727729
sqlExecute(
728730
"UPDATE subscriptions SET enabled=1 WHERE label=? AND address=?",
729731
subscriptions[subcur][0],
730732
subscriptions[subcur][1])
731-
shared.reloadBroadcastSendersForWhichImWatching()
733+
helper_addressbook.reloadBroadcastSendersForWhichImWatching()
732734
subscriptions[subcur][2] = True
733735
elif t == "4":
734736
sqlExecute(
735737
"UPDATE subscriptions SET enabled=0 WHERE label=? AND address=?",
736738
subscriptions[subcur][0],
737739
subscriptions[subcur][1])
738-
shared.reloadBroadcastSendersForWhichImWatching()
740+
helper_addressbook.reloadBroadcastSendersForWhichImWatching()
739741
subscriptions[subcur][2] = False
740742
elif menutab == 6:
741743
set_background_title(d, "Address Book Dialog Box")
@@ -764,7 +766,7 @@ def handlech(c, stdscr):
764766
subscriptions.reverse()
765767

766768
sqlExecute("INSERT INTO subscriptions VALUES (?,?,?)", label, addr, True)
767-
shared.reloadBroadcastSendersForWhichImWatching()
769+
helper_addressbook.reloadBroadcastSendersForWhichImWatching()
768770
elif t == "3":
769771
r, t = d.inputbox("Input new address")
770772
if r == d.DIALOG_OK:

src/bitmessagemain.py

Lines changed: 12 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
app_dir = pathmagic.setup()
2020

2121
import depends
22+
2223
depends.check_dependencies()
2324

2425
import getopt
@@ -30,25 +31,25 @@
3031
import traceback
3132

3233
import defaults
34+
import helper_addressbook
3335
import shared
3436
import shutdown
3537
import state
3638
from bmconfigparser import BMConfigParser
3739
from debug import logger # this should go before any threads
38-
from helper_startup import (
39-
adjustHalfOpenConnectionsLimit, fixSocket, start_proxyconfig)
40+
from helper_startup import (adjustHalfOpenConnectionsLimit, fixSocket,
41+
start_proxyconfig)
4042
from inventory import Inventory
4143
# Network objects and threads
42-
from network import (
43-
BMConnectionPool, Dandelion, AddrThread, AnnounceThread, BMNetworkThread,
44-
InvThread, ReceiveQueueThread, DownloadThread, UploadThread
45-
)
44+
from network import (AddrThread, AnnounceThread, BMConnectionPool,
45+
BMNetworkThread, Dandelion, DownloadThread, InvThread,
46+
ReceiveQueueThread, UploadThread)
4647
from network.knownnodes import readKnownNodes
4748
from singleinstance import singleinstance
49+
from storage import filesystem
4850
# Synchronous threads
49-
from threads import (
50-
set_thread_name, printLock,
51-
addressGenerator, objectProcessor, singleCleaner, singleWorker, sqlThread)
51+
from threads import (addressGenerator, objectProcessor, printLock,
52+
set_thread_name, singleCleaner, singleWorker, sqlThread)
5253

5354

5455
def signal_handler(signum, frame):
@@ -235,8 +236,8 @@ def start(self):
235236

236237
# Not needed if objproc disabled
237238
if state.enableObjProc:
238-
shared.reloadMyAddressHashes()
239-
shared.reloadBroadcastSendersForWhichImWatching()
239+
filesystem.reloadMyAddressHashes()
240+
helper_addressbook.reloadBroadcastSendersForWhichImWatching()
240241

241242
# API is also objproc dependent
242243
if config.safeGetBoolean('bitmessagesettings', 'apienabled'):

0 commit comments

Comments
 (0)