Add logic

src/registrar/models/user.py

@@ -424,3 +424,8 @@ class User(AbstractUser):
     def is_org_user(self, request):
         has_organization_feature_flag = flag_is_active(request, "organization_feature")
         return has_organization_feature_flag and self.has_base_portfolio_permission()
+    
+    def user_domain_count(self):
+        """Returns the number of domains associated with this user through UserDomainRole"""
+        available_domains = UserDomainRole.objects.filter(user=self, domain__isnull=False)
+        return available_domains.count()

src/registrar/templates/home.html

@@ -29,7 +29,7 @@
       </a>
     </p>
 
-    {% include "includes/domains_table.html" %}
+    {% include "includes/domains_table.html" user_domain_count=user_domain_count %}
     {% include "includes/domain_requests_table.html" %}
 
   </div>

src/registrar/templates/includes/domains_table.html

@@ -35,6 +35,7 @@
           </form>
         </section>
       </div>
+      {% if user_domains_count and user_domains_count > 0 %}
       <div class="section--outlined__utility-button mobile-lg:padding-right-105 {% if portfolio %} mobile:grid-col-12 desktop:grid-col-6 desktop:padding-left-3{% endif %}">
         <section aria-label="Domains report component" class="mobile-lg:margin-top-205">
           <a href="{% url 'export_data_type_user' %}" class="usa-button usa-button--unstyled" role="button">
@@ -44,6 +45,7 @@
           </a>
         </section>
       </div>
+      {% endif %}
     </div>
     <!-- Use portfolio_base_permission when merging into 2366 then delete this comment -->
     {% if portfolio %}

src/registrar/templates/portfolio_domains.html

@@ -6,5 +6,5 @@
 
 {% block portfolio_content %}
     <h1 id="domains-header">Domains</h1>
-    {% include "includes/domains_table.html" with portfolio=portfolio %}
+    {% include "includes/domains_table.html" with portfolio=portfolio user_domain_count=user_domain_count %}
 {% endblock %}

src/registrar/views/index.py

@@ -8,5 +8,6 @@ def index(request):
     if request.user.is_authenticated:
         # This controls the creation of a new domain request in the wizard
         request.session["new_request"] = True
+        context["user_domain_count"] = request.user.user_domain_count()
 
     return render(request, "home.html", context)

src/registrar/views/portfolios.py

@@ -29,6 +29,7 @@ class PortfolioDomainsView(PortfolioDomainsPermissionView, View):
             context["has_organization_feature_flag"] = flag_is_active(request, "organization_feature")
             portfolio = get_object_or_404(Portfolio, id=portfolio_id)
             context["portfolio"] = portfolio
+            context["user_domain_count"] = self.request.user.user_domain_count()
 
         return render(request, "portfolio_domains.html", context)