commit f9274e65df602d698b7d646cf17d7c452b248e43 · edavis.dev/bsky-tools

+8

feed_manager.py

···

       1
       1
       +
       from feeds.popular import PopularFeed

     

       2
       2
       +
       from feeds.rapidfire import RapidFireFeed

     

       3
       3
       +
       

     

       1
       4
        
       class FeedManager:

     

       2
       5
        
           def __init__(self):

     

       3
       6
        
               self.feeds = {}

     
···

       22
       25
        
           def run_tasks_minute(self):

     

       23
       26
        
               for feed in self.feeds.values():

     

       24
       27
        
                   feed.run_tasks_minute()

     

       28
       28
       +
       

     

       29
       29
       +
       

     

       30
       30
       +
       manager = FeedManager()

     

       31
       31
       +
       # manager.register(PopularFeed)

     

       32
       32
       +
       manager.register(RapidFireFeed)

+1 -7

feedgen.py

···

       9
       9
        
       import dag_cbor

     

       10
       10
        
       import websockets

     

       11
       11
        
       

     

       12
       12
       -
       from feed_manager import FeedManager

     

       13
       13
       -
       from feeds.rapidfire import RapidFireFeed

     

       14
       14
       -
       from feeds.popular import PopularFeed

     

       12
       12
       +
       from feed_manager import manager as feed_manager

     

       15
       13
        
       from firehose_manager import FirehoseManager

     

       16
       14
        
       

     

       17
       15
        
       logging.basicConfig(

     
···

       58
       56
        
       

     

       59
       57
        
       async def main():

     

       60
       58
        
           firehose_manager = FirehoseManager()

     

       61
       61
       -
       

     

       62
       62
       -
           feed_manager = FeedManager()

     

       63
       63
       -
           feed_manager.register(RapidFireFeed)

     

       64
       64
       -
           # feed_manager.register(PopularFeed)

     

       65
       59
        
       

     

       66
       60
        
           current_minute = None

     

       67
       61
        
           async for commit in firehose_events(firehose_manager):

+3 -7

feedweb.py

···

       2
       2
        
       

     

       3
       3
        
       from flask import Flask, request, jsonify

     

       4
       4
        
       

     

       5
       5
       -
       from feed_manager import FeedManager

     

       5
       5
       +
       from feed_manager import manager as feed_manager

     

       6
       6
        
       from feeds.rapidfire import RapidFireFeed

     

       7
       7
        
       from feeds.popular import PopularFeed

     

       8
       8
        
       

     
···

       10
       10
        
       

     

       11
       11
        
       @app.route('/xrpc/app.bsky.feed.getFeedSkeleton')

     

       12
       12
        
       def get_feed_skeleton():

     

       13
       13
       -
           manager = FeedManager()

     

       14
       14
       -
           manager.register(RapidFireFeed)

     

       15
       15
       -
           # manager.register(PopularFeed)

     

       16
       16
       -
       

     

       17
       13
        
           try:

     

       18
       14
        
               limit = int(request.args.get('limit', 50))

     

       19
       15
        
           except ValueError:

     
···

       33
       29
        
       

     

       34
       30
        
           if request.args.get('debug', '0') == '1':

     

       35
       31
        
               headers = {'Content-Type': 'text/plain; charset=utf-8'}

     

       36
       36
       -
               debug = manager.serve_feed_debug(feed_uri, limit, offset, langs)

     

       32
       32
       +
               debug = feed_manager.serve_feed_debug(feed_uri, limit, offset, langs)

     

       37
       33
        
               return debug, headers

     

       38
       34
        
       

     

       39
       39
       -
           posts = manager.serve_feed(feed_uri, limit, offset, langs)

     

       35
       35
       +
           posts = feed_manager.serve_feed(feed_uri, limit, offset, langs)

     

       40
       36
        
           offset += len(posts)

     

       41
       37
        
       

     

       42
       38
        
           return dict(cursor=str(offset), feed=[dict(post=uri) for uri in posts])