commit 9be0830977766eec37d5d9c7ce1ea4efd9258b4b · edavis.dev/bsky-tools

+1 -1

bsky-activity.py

···

       45
       45
        
           sys.stdout.flush()

     

       46
       46
        
       

     

       47
       47
        
           op_count = 0

     

       48
       48
       -
           for commit, op in subscribe_commits():

     

       48
       48
       +
           for commit, op in subscribe_commits(redis_cnx):

     

       49
       49
        
               if op['action'] != 'create':

     

       50
       50
        
                   continue

     

       51
       51

+3 -2

firehose_utils.py

···

       3
       3
        
       from atproto import CAR

     

       4
       4
        
       from io import BytesIO

     

       5
       5
        
       

     

       6
       6
       -
       def subscribe_commits():

     

       7
       7
       -
           redis_cnx = redis.Redis()

     

       6
       6
       +
       def subscribe_commits(redis_cnx=None):

     

       7
       7
       +
           if redis_cnx is None:

     

       8
       8
       +
               redis_cnx = redis.Redis()

     

       8
       9
        
           redis_sub = redis_cnx.pubsub(ignore_subscribe_messages=True)

     

       9
       10
        
           redis_sub.subscribe('bsky-tools:firehose:stream')

     

       10
       11