Merge pull request #86 from nautobot/main-data

Add load-mock-data command

README.md

@@ -57,6 +57,20 @@ The container can be built locally, if preferred.
 docker run -itd --name nautobot -p 8000:8000 nautobot-lab
 ```
 
+### Importing the demo dataset
+
+The nautobot-lab container provides demo dataset similar to the Nautobot public sandbox (http://demo.nautobot.com).
+This dataset can be used to quickly populate your Nautobot instance with the real life data! You can use it to view data, experiment or perform experimental or development work.
+
+Use following command to import the data into your lab container:
+```shell
+docker exec -itd nautobot load-mock-data
+```
+
+!!! warning
+    **Importing dataset will erase your existing data**. Issuing the `load-mock-data` will result in destroying your existing dataset and importing new one.
+    **Dataset contains default credentials**. If you decide to import provided dataset, the dataset credentials will be used: username: `admin`, password: `admin`.
+
 ## The Default Account
 
 The nautobot-lab container comes with a user pre-defined. The username is `demo` and its corresponding password is `nautobot`. If you choose to build your own container, you can define your own attributes for the default account by setting the following environment variables before executing the  `docker build` command.

pb_nautobot_install.yml

@@ -130,6 +130,14 @@
         mode: u=rwX,g=rX,o=rX
         recurse: "yes"
 
+    - name: "ADD LOAD MOCK DATA SCRIPT"
+      ansible.builtin.template:
+        src: "loaddata.sh"
+        dest: "/usr/local/bin/load-mock-data"
+        owner: "nautobot"
+        group: "nautobot"
+        mode: "0755"
+
     - name: "UPDATE REDIS.CONF"
       ansible.builtin.template:
         src: "redis.conf"

templates/loaddata.sh

@@ -0,0 +1,18 @@
+#!/bin/bash
+
+export PGUSER="postgres"
+export PGDATABASE=$DB_NAME
+
+psql -c "DROP SCHEMA public CASCADE";
+psql -c "CREATE SCHEMA public";
+psql -c "GRANT ALL ON SCHEMA public TO postgres";
+psql -c "GRANT ALL ON SCHEMA public TO public";
+psql -c "GRANT ALL ON SCHEMA public TO ${DB_USER}";
+
+export PGUSER=$DB_USER
+export PGPASSWORD=$DB_PASSWORD
+
+psql < {{ nautobot_root }}/templates/nautobot_backup.dump
+
+{{ nautobot_root }}/bin/nautobot-server migrate
+{{ nautobot_root }}/bin/nautobot-server invalidate all