Notify.php

diff --git a/Notify.php b/Notify.php
index cd24525..e794bc8 100644 (file)
--- a/Notify.php
+++ b/Notify.php
@@ -307,6 +307,14 @@ class Pman_Core_Notify extends Pman
         
     }
 
+    function setCompany() {
+        $co = DB_DataObject::Factory('Companies');
+        if(!$co->get('code', 'MediaWatchMe')) {
+            $this->errors[] = 'Invalid Supplier Code: MediaWatchMe';
+        }
+        $this->company = $co->id;
+    }
+
     function output()
     {
         die("Done\n");