From 0606bb0d4c8a90c161d518034e691153f9856f33 Mon Sep 17 00:00:00 2001
From: pepperpepperpepper <pepper@scannerjammer.com>
Date: Mon, 7 Dec 2015 20:46:41 -0800
Subject: added query

---
 ricky/pb.py | 22 ++++++++++++++++++++--
 1 file changed, 20 insertions(+), 2 deletions(-)

(limited to 'ricky/pb.py')

diff --git a/ricky/pb.py b/ricky/pb.py
index 2b07737..10674f2 100644
--- a/ricky/pb.py
+++ b/ricky/pb.py
@@ -1,3 +1,4 @@
+import os
 import urllib
 import urllib2
 import sys
@@ -41,9 +42,9 @@ class Pb(object):
     def call(self, params):
         if self._offline:
             sys.path.append("./photoblaster")
-            from photoblaster.modules import Pb
+            from photoblaster.modules import Pb as _Pb
             from photoblaster.config import LOCAL as PBLOCAL
-            for pbcls in Pb.__subclasses__():
+            for pbcls in _Pb.__subclasses__():
                 if pbcls.__name__ == self.__class__.__name__:
                     params_dict = params.as_dict()
                     instance = pbcls(**params_dict)
@@ -54,3 +55,20 @@ class Pb(object):
         return json.loads(
             self.post_request(self.url, params.as_dict())
         )
+
+    def params_from_url(self, url):
+        """
+        retrieves image params from db using the url
+        """
+        if self._offline:
+            sys.path.append("./photoblaster")
+            from photoblaster.db.models.imcmd import ImCmd
+            newfile = os.path.split(url)[-1]
+            files = ImCmd.search({"newfile": newfile})
+            if not len(files):
+                return None
+            filedata = json.loads(files[0].dataobj)
+            return filedata
+        else:
+            # needs a route
+            raise NotImplementedError("Not yet implemented\n")
-- 
cgit v1.2.3-70-g09d2