Updated generator to parse out attachment and PDF capable models

generator/generate

@@ -439,6 +439,11 @@ foreach($scrape_apis as $scrape_api) {
                     //Debug
                     $model->printPropertyTable();
                 }
+
+                if(strpos($primary_model->rawHTML, 'returned as PDF') !== false){
+                    //Assume that it does support it.
+                    $primary_model->setSupportsPDF(true);
+                }
             }
 
         });

generator/objects/Model.php

@@ -14,7 +14,9 @@ class Model {
     private $api;
 
     /**
-     * @var The parent model of the object.  This is if it's defined as a secondary model on a doc page
+     * The parent model of the object.  This is if it's defined as a secondary model on a doc page
+     *
+     * @var Model $parent_model
      */
     private $parent_model;
 
@@ -23,6 +25,14 @@ class Model {
      */
     private $guid_property;
 
+    /**
+     * @var bool $supports_pdf
+     */
+    private $supports_pdf;
+
+
+    public $rawHTML;
+
     /**
      * No args in constructor.  Most things are not known when it's built
      */
@@ -31,6 +41,7 @@ public function __construct(){
         $this->properties = array();
         $this->methods = array();
         $this->sub_models = array();
+        $this->supports_pdf = false;
     }
 
     /**
@@ -250,6 +261,31 @@ public function setUrl($url) {
         $this->url = $url;
     }
 
+
+    /**
+     * This should be enough to tell if it supports them
+     *
+     * @return boolean
+     */
+    public function getSupportsAttachments() {
+        return $this->hasProperty('HasAttachments');
+    }
+
+    /**
+     * @return boolean
+     */
+    public function getSupportsPDF() {
+        return $this->supports_pdf;
+    }
+
+    /**
+     * @param boolean $supports_pdf
+     */
+    public function setSupportsPDF($supports_pdf) {
+        $this->supports_pdf = $supports_pdf;
+    }
+
+
     /**
      * Shortcut for getting them from $api where the group is the same as this name
      *

generator/templates/model.twig

@@ -4,12 +4,26 @@ namespace XeroPHP\Models\{{ model.Namespace }};
 
 use XeroPHP\Remote;
 
+{% if model.SupportsPDF %}
+use XeroPHP\Traits\PDFTrait;
+{% endif %}
+{% if model.SupportsAttachments %}
+use XeroPHP\Traits\AttachmentTrait;
+
+{% endif %}
 {% for class in model.UsedClasses %}
 use XeroPHP\Models\{{ class }};
 {% endfor %}
 
 class {{ model.ClassName }} extends Remote\Object {
 
+{% if model.SupportsPDF %}
+    use PDFTrait;
+{% endif %}
+{% if model.SupportsAttachments %}
+    use AttachmentTrait;
+
+{% endif %}
 {% for property in model.Properties %}
     /**
      * {{ property.Description|wordwrap(100, "\n     * ")|raw }}