diff --git a/r2/r2/lib/promote.py b/r2/r2/lib/promote.py index eceb987e0..6679eb753 100644 --- a/r2/r2/lib/promote.py +++ b/r2/r2/lib/promote.py @@ -64,6 +64,7 @@ from r2.models import ( PromotedLink, PromotionLog, PromotionWeights, + PromotedLinkRoadblock, Subreddit, ) from r2.models.keyvalue import NamedGlobals @@ -223,6 +224,14 @@ def get_roadblocks(): blobs.sort(key=lambda x: x[1]) return blobs + +def migrate_roadblocks(): + q = get_roadblocks() + for sr_name, start, end in q: + sr = Subreddit._by_name(sr_name) + PromotedLinkRoadblock.add(sr, start, end) + + # control functions class RenderableCampaign():