From d204066990931f131043eecb42dc551c4454222f Mon Sep 17 00:00:00 2001
From: Stefano Alberto Russo <stefano.russo@gmail.com>
Date: Wed, 20 Oct 2021 19:21:16 +0200
Subject: [PATCH] Added support for genrating ORM model relationships graphs.

---
 services/webapp/Dockerfile               | 13 ++++++++-----
 services/webapp/code/rosetta/settings.py |  3 ++-
 2 files changed, 10 insertions(+), 6 deletions(-)

diff --git a/services/webapp/Dockerfile b/services/webapp/Dockerfile
index dff25b2..1c0c880 100644
--- a/services/webapp/Dockerfile
+++ b/services/webapp/Dockerfile
@@ -63,6 +63,14 @@ COPY db_conf.sh /db_conf.sh
 RUN mkdir /var/log/webapp/ && chown rosetta:rosetta /var/log/webapp/
 
 
+#------------------------------
+# Viz
+#------------------------------
+RUN apt install python-pygraphviz graphviz-dev -y
+RUN pip3 install django-extensions pygraphviz
+# Example usage: rosetta/shell webapp "cd /opt/code && python3 manage.py graph_models core_app --exclude-models LoginToken,Text -o ORM.png"
+
+
 #------------------------------
 # Supervisord
 #------------------------------
@@ -81,8 +89,3 @@ COPY prestartup_webapp.sh /prestartup/
 RUN touch -m /prestartup/prestartup_webapp.sh
 
 
-
-
-
-
-
diff --git a/services/webapp/code/rosetta/settings.py b/services/webapp/code/rosetta/settings.py
index e3badd8..436f9d8 100644
--- a/services/webapp/code/rosetta/settings.py
+++ b/services/webapp/code/rosetta/settings.py
@@ -42,7 +42,8 @@ INSTALLED_APPS = [
     'django.contrib.messages',
     'django.contrib.staticfiles',
     'rest_framework',
-    'rest_framework_swagger',  
+    'rest_framework_swagger',
+    'django_extensions', 
 ]
 
 MIDDLEWARE = [
-- 
GitLab