From 201d0d05a99bdef218ad59039841d62e404086ea Mon Sep 17 00:00:00 2001
From: Brian Simpson <bsimpson63@gmail.com>
Date: Mon, 12 Aug 2013 18:56:55 -0400
Subject: [PATCH] Migrate roadblocks to Cassandra.

---
 r2/r2/lib/promote.py | 9 +++++++++
 1 file changed, 9 insertions(+)

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():