From 67dbe02deb6012891d9c6609f003d1a8cb8150ac Mon Sep 17 00:00:00 2001
From: Jeremy Stretch <jstretch@digitalocean.com>
Date: Thu, 12 Jul 2018 16:55:23 -0400
Subject: [PATCH] Enabled export templates for virtual chassis

---
 netbox/dcim/models.py                          | 7 +++++++
 netbox/extras/constants.py                     | 2 +-
 netbox/templates/dcim/virtualchassis_list.html | 4 ++++
 3 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/netbox/dcim/models.py b/netbox/dcim/models.py
index 0eea17c32..9a1f71847 100644
--- a/netbox/dcim/models.py
+++ b/netbox/dcim/models.py
@@ -2267,6 +2267,7 @@ class VirtualChassis(ChangeLoggedModel):
     tags = TaggableManager()
 
     serializer = 'dcim.api.serializers.VirtualChassisSerializer'
+    csv_headers = ['master', 'domain']
 
     class Meta:
         ordering = ['master']
@@ -2286,3 +2287,9 @@ class VirtualChassis(ChangeLoggedModel):
             raise ValidationError({
                 'master': "The selected master is not assigned to this virtual chassis."
             })
+
+    def to_csv(self):
+        return (
+            self.master,
+            self.domain,
+        )
diff --git a/netbox/extras/constants.py b/netbox/extras/constants.py
index 2663d81c3..e1e12b78f 100644
--- a/netbox/extras/constants.py
+++ b/netbox/extras/constants.py
@@ -50,7 +50,7 @@ GRAPH_TYPE_CHOICES = (
 EXPORTTEMPLATE_MODELS = [
     'provider', 'circuit',                                                          # Circuits
     'site', 'region', 'rack', 'rackgroup', 'manufacturer', 'devicetype', 'device',  # DCIM
-    'consoleport', 'powerport', 'interfaceconnection',                              # DCIM
+    'consoleport', 'powerport', 'interfaceconnection', 'virtualchassis',            # DCIM
     'aggregate', 'prefix', 'ipaddress', 'vlan', 'vrf',                              # IPAM
     'tenant',                                                                       # Tenancy
     'cluster', 'virtualmachine',                                                    # Virtualization
diff --git a/netbox/templates/dcim/virtualchassis_list.html b/netbox/templates/dcim/virtualchassis_list.html
index 756bc5f28..3f1781d95 100644
--- a/netbox/templates/dcim/virtualchassis_list.html
+++ b/netbox/templates/dcim/virtualchassis_list.html
@@ -1,7 +1,11 @@
 {% extends '_base.html' %}
+{% load buttons %}
 {% load helpers %}
 
 {% block content %}
+<div class="pull-right">
+    {% export_button content_type %}
+</div>
 <h1>{% block title %}Virtual Chassis{% endblock %}</h1>
 <div class="row">
 	<div class="col-md-9">