superuser sees all - #347

Author: carlosmouracorreia

rootio/configuration/views.py

@@ -19,6 +19,7 @@
 from ..radio.models import Station, Network
 from ..user.models import User
 from ..utils import upload_to_s3, make_dir, save_uploaded_file, jquery_dt_paginator
+from rootio.user import ADMIN
 
 configuration = Blueprint('configuration', __name__, url_prefix='/configuration')
 
@@ -27,7 +28,10 @@
 @login_required
 def index():
     # get all the user's networks and their stations
-    networks = Network.query.outerjoin(Station).join(User, Network.networkusers).filter(User.id == current_user.id).all()
+    if current_user.role_code == ADMIN:
+        networks = Network.query.outerjoin(Station).join(User, Network.networkusers).all()
+    else:
+        networks = Network.query.outerjoin(Station).join(User, Network.networkusers).filter(User.id == current_user.id).all()
     stations = []
 
     for network in networks:
@@ -76,7 +80,7 @@ def tts():
 @configuration.route('/telephony/', methods=['GET', 'POST'])
 @login_required
 def telephony():
-    stations = Station.query.join(Network).join(User, Network.networkusers).filter(User.id == current_user.id).all()
+    stations = Station.get_stations(current_user)
     return render_template('configuration/stations_telephony.html', stations=stations)
 
 
@@ -138,7 +142,7 @@ def sip_configuration(station_id):
 @configuration.route('/sip_telephony', methods=['GET', 'POST'])
 @login_required
 def sip_telephony():
-    stations = Station.query.join(Network).join(User, Network.networkusers).filter(User.id == current_user.id).all()
+    stations = Station.get_stations(current_user)
     return render_template('configuration/sip_telephony.html', stations=stations)
 
 
@@ -161,14 +165,14 @@ def station_audio_level(station_id):
 @configuration.route('/station_audio_levels', methods=['GET'])
 @login_required
 def station_audio_levels():
-    stations = Station.query.join(Network).join(User, Network.networkusers).filter(User.id == current_user.id).all()
+    stations = Station.get_stations(current_user)
     return render_template('configuration/station_audio_levels.html', stations=stations, active='stations')
 
 
 @configuration.route('/ivr_menus', methods=['GET', 'POST'])
 @login_required
 def ivr_menus():
-    stations = Station.query.join(Network).join(User, Network.networkusers).filter(User.id == current_user.id).all()
+    stations = Station.get_stations(current_user)
 
     station_data = []
     for station in stations:
@@ -270,7 +274,7 @@ def ivr_menu():
 @configuration.route('/tts_settings', methods=['GET', 'POST'])
 @login_required
 def tts_settings():
-    stations = Station.query.join(Network).join(User, Network.networkusers).filter(User.id == current_user.id).all()
+    stations = Station.get_stations(current_user)
     return render_template('configuration/tts_settings.html', stations=stations)
 
 
@@ -293,7 +297,7 @@ def tts_setting(station_id):
 @configuration.route('/synchronization_settings', methods=['GET', 'POST'])
 @login_required
 def synchronization_settings():
-    stations = Station.query.join(Network).join(User, Network.networkusers).filter(User.id == current_user.id).all()
+    stations = Station.get_stations(current_user)
     return render_template('configuration/synchronization_settings.html', stations=stations)
 
 
@@ -316,7 +320,7 @@ def synchronization_setting(station_id):
 @configuration.route('/content', methods=['GET'])
 @login_required
 def content_settings():
-    stations = Station.query.join(Network).join(User, Network.networkusers).filter(User.id == current_user.id).all()
+    stations = Station.get_stations(current_user)
     return render_template('configuration/content_settings.html', stations=stations)
 
 
@@ -339,7 +343,7 @@ def content_setting(station_id):
 @configuration.route('/voice_prompts', methods=['GET', 'POST'])
 @login_required
 def voice_prompts():
-    stations = Station.query.join(Network).join(User, Network.networkusers).filter(User.id == current_user.id).all()
+    stations = Station.get_stations(current_user)
 
     station_data = []
     for station in stations:

rootio/content/views.py

@@ -18,6 +18,7 @@
 from ..radio.models import ContentType, Person, Network, Station
 from ..user.models import User
 from ..utils import jquery_dt_paginator, upload_to_s3, make_dir, save_uploaded_file
+from rootio.user import ADMIN
 
 ALLOWED_EXTENSIONS = set(['wav', 'mp3'])
 
@@ -283,7 +284,10 @@ def track_toggle_play(track_id):
 @content.route('/hosts/')
 @login_required
 def list_hosts():
-    hosts = Person.query.join(Person, Network.people).join(User, Network.networkusers).filter(
+    if current_user.role_code == ADMIN:
+        hosts = Person.query.join(Person, Network.people).join(User, Network.networkusers).all()
+    else:
+        hosts = Person.query.join(Person, Network.people).join(User, Network.networkusers).filter(
         User.id == current_user.id).all()
     return render_template('content/content_hosts.html', hosts=hosts)
 
@@ -363,7 +367,10 @@ def host_delete(host_id):
 @content.route('/community_content/', methods=['GET', 'POST'])
 @login_required
 def community_content():
-    community_contents = CommunityContent.query.join(Station).join(Network).join(User, Network.networkusers).filter(
+    if current_user.role_code == ADMIN:
+        community_contents = CommunityContent.query.join(Station).join(Network).join(User, Network.networkusers).all()
+    else:
+        community_contents = CommunityContent.query.join(Station).join(Network).join(User, Network.networkusers).filter(
         User.id == current_user.id).all()
     return render_template('content/community_content.html', community_contents=community_contents)
 

rootio/radio/models.py

@@ -157,6 +157,15 @@ class Station(BaseMixin, db.Model):
     broadcast_ip = db.Column(db.String(16))
     broadcast_port = db.Column(db.String(16))
 
+    @classmethod
+    def get_stations(cls, current_user):
+        from rootio.user import ADMIN
+        from ..user.models import User
+        
+        if current_user.role_code == ADMIN:
+            return Station.query.join(Network).join(User, Network.networkusers).all()
+        else:
+            return Station.query.join(Network).join(User, Network.networkusers).filter(User.id == current_user.id).all()
     def init(self):
         # load dummy program
         # init state machine

rootio/radio/views.py

@@ -31,6 +31,7 @@
 from ..extensions import db, csrf
 from ..user.models import User, RootioUser
 from ..utils import error_dict, fk_lookup_form_data, format_log_line, events_action_display_map
+from rootio.user import ADMIN
 
 radio = Blueprint('radio', __name__, url_prefix='/radio')
 
@@ -39,7 +40,10 @@
 @login_required
 def index():
     # get all the user's networks and their stations
-    networks = Network.query.outerjoin(Station).join(User, Network.networkusers).filter(User.id == current_user.id).all()
+    if current_user.role_code == ADMIN:
+        networks = Network.query.outerjoin(Station).join(User, Network.networkusers).all()
+    else:
+        networks = Network.query.outerjoin(Station).join(User, Network.networkusers).filter(User.id == current_user.id).all()
     return render_template('radio/index.html', networks=networks, userid=current_user.id, now=datetime.now)
 
 
@@ -83,7 +87,7 @@ def network_add():
 @radio.route('/station/', methods=['GET'])
 @login_required
 def list_stations():
-    stations = Station.query.join(Network).join(User, Network.networkusers).filter(User.id == current_user.id).all()
+    stations = Station.get_stations(current_user)
     # stations = Station.query.join(Network).join(User).filter(User.id == current_user.id).all()
     return render_template('radio/stations.html', stations=stations, active='stations')
 
@@ -134,7 +138,10 @@ def station_add():
 @radio.route('/program/', methods=['GET'])
 @login_required
 def programs():
-    programs = Program.query.join(Program, Network.programs).join(User, Network.networkusers).filter(
+    if current_user.role_code == ADMIN:
+        programs = Program.query.join(Program, Network.programs).join(User, Network.networkusers).filter(Program.program_type_id != 2).all()
+    else:
+        programs = Program.query.join(Program, Network.programs).join(User, Network.networkusers).filter(
         User.id == current_user.id).filter(Program.program_type_id != 2).all()
     return render_template('radio/programs.html', programs=programs, active='programs')
 
@@ -146,12 +153,22 @@ def program_definition(program_id):
     # form = ProgramForm(obj=program, program_structure="test", next=request.args.get('next'))
 
     # hosts in my network
-    hosts = Person.query.join(Person, Network.people).join(User, Network.networkusers).filter(
+    if current_user.role_code == ADMIN:
+        hosts = Person.query.join(Person, Network.people).join(User, Network.networkusers).all()
+    else:
+        hosts = Person.query.join(Person, Network.people).join(User, Network.networkusers).filter(
         User.id == current_user.id).all()
     news = ContentTrack.query.join(ContentType).filter(ContentType.name == "News").all()
     ads = ContentTrack.query.join(ContentType).filter(ContentType.name == "Advertisements").all()
 
-    medias = ContentTrack.query.join(User, Network.networkusers)\
+    if current_user.role_code == ADMIN:
+        medias = ContentTrack.query.join(User, Network.networkusers)\
+        .join(ContentTrack, ContentType)\
+        .filter(ContentType.name == "Media")\
+        .filter(ContentTrack.deleted != True)\
+        .all()
+    else:
+        medias = ContentTrack.query.join(User, Network.networkusers)\
         .filter(User.id == current_user.id)\
         .join(ContentTrack, ContentType)\
         .filter(ContentType.name == "Media")\
@@ -197,34 +214,59 @@ def program_add():
     form = ProgramForm(request.form)
     program = None
 
-    # hosts in my network
-    hosts = Person.query.join(Person, Network.people).join(User, Network.networkusers)\
-                                                     .filter(User.id == current_user.id)\
-                                                     .all()
-    news = ContentTrack.query.join(User, Network.networkusers)\
-                             .filter(User.id == current_user.id)\
-                             .join(ContentTrack, ContentType)\
-                             .filter(ContentType.name == "News")\
-                             .filter(ContentTrack.deleted != True)\
-                             .all()
-    ads = ContentTrack.query.join(User, Network.networkusers)\
-                            .filter(User.id == current_user.id)\
-                            .join(ContentTrack, ContentType)\
-                            .filter(ContentType.name == "Advertisements")\
-                            .filter(ContentTrack.deleted != True)\
-                            .all()
-      
-    medias = ContentTrack.query.join(User, Network.networkusers)\
-                               .filter(User.id == current_user.id)\
-                               .join(ContentTrack, ContentType)\
-                               .filter(ContentType.name == "Media")\
-                               .filter(ContentTrack.deleted != True)\
-                               .all()
-
-
-    podcasts = ContentPodcast.query.join(User, Network.networkusers)\
-                                   .filter(User.id == current_user.id)\
-                                   .all()
+    if current_user.role_code == ADMIN:
+         # hosts in my network
+        hosts = Person.query.join(Person, Network.people).join(User, Network.networkusers)\
+                                                        .all()
+        news = ContentTrack.query.join(User, Network.networkusers)\
+                                .join(ContentTrack, ContentType)\
+                                .filter(ContentType.name == "News")\
+                                .filter(ContentTrack.deleted != True)\
+                                .all()
+        ads = ContentTrack.query.join(User, Network.networkusers)\
+                                .join(ContentTrack, ContentType)\
+                                .filter(ContentType.name == "Advertisements")\
+                                .filter(ContentTrack.deleted != True)\
+                                .all()
+        
+        medias = ContentTrack.query.join(User, Network.networkusers)\
+                                .join(ContentTrack, ContentType)\
+                                .filter(ContentType.name == "Media")\
+                                .filter(ContentTrack.deleted != True)\
+                                .all()
+
+
+        podcasts = ContentPodcast.query.join(User, Network.networkusers)\
+                                    .all()
+    else:
+        # hosts in my network
+        hosts = Person.query.join(Person, Network.people).join(User, Network.networkusers)\
+                                                        .filter(User.id == current_user.id)\
+                                                        .all()
+        news = ContentTrack.query.join(User, Network.networkusers)\
+                                .filter(User.id == current_user.id)\
+                                .join(ContentTrack, ContentType)\
+                                .filter(ContentType.name == "News")\
+                                .filter(ContentTrack.deleted != True)\
+                                .all()
+        ads = ContentTrack.query.join(User, Network.networkusers)\
+                                .filter(User.id == current_user.id)\
+                                .join(ContentTrack, ContentType)\
+                                .filter(ContentType.name == "Advertisements")\
+                                .filter(ContentTrack.deleted != True)\
+                                .all()
+        
+        medias = ContentTrack.query.join(User, Network.networkusers)\
+                                .filter(User.id == current_user.id)\
+                                .join(ContentTrack, ContentType)\
+                                .filter(ContentType.name == "Media")\
+                                .filter(ContentTrack.deleted != True)\
+                                .all()
+
+
+        podcasts = ContentPodcast.query.join(User, Network.networkusers)\
+                                    .filter(User.id == current_user.id)\
+                                    .all()
     community_contents = {"data": [{"type": "Ads", "category_id": "1"}, {"type": "Announcements", "category_id": "2"},
                                    {"type": "Greetings", "category_id": "3"}]}
 
@@ -265,7 +307,10 @@ def program_delete(program_id):
 @radio.route('/music_program/', methods=['GET'])
 @login_required
 def list_music_programs():
-    music_programs = Program.query.join(Program, Network.programs).join(User, Network.networkusers).filter(
+    if current_user.role_code == ADMIN:
+        music_programs = Program.query.join(Program, Network.programs).join(User, Network.networkusers).filter(Program.program_type_id == 2).all()
+    else:
+        music_programs = Program.query.join(Program, Network.programs).join(User, Network.networkusers).filter(
         User.id == current_user.id).filter(Program.program_type_id == 2).all()
     return render_template('radio/music_programs.html', music_programs=music_programs, active='programs')
 
@@ -276,11 +321,15 @@ def music_program_add():
     form = ProgramForm(request.form)
     program = None
 
-    # Playlists and streams in my network
-    playlists = ContentMusicPlaylist.query.join(Station).join(Network).join(User, Network.networkusers).filter(
-        User.id == current_user.id).filter(ContentMusicPlaylist.deleted != True).all()  # Playlist->Station->Network->user
-    streams = ContentStream.query.join(User, Network.networkusers).filter(
-        User.id == current_user.id).all()  # created by -> user -> Network
+    if current_user.role_code == ADMIN:
+        playlists = ContentMusicPlaylist.query.join(Station).join(Network).join(User, Network.networkusers).filter(ContentMusicPlaylist.deleted != True).all()  # Playlist->Station->Network->user
+        streams = ContentStream.query.join(User, Network.networkusers).filter(
+            User.id == current_user.id).all()  # created by -> user -> Network
+    else:
+        # Playlists and streams in my network
+        playlists = ContentMusicPlaylist.query.join(Station).join(Network).join(User, Network.networkusers).filter(ContentMusicPlaylist.deleted != True).all()  # Playlist->Station->Network->user
+        streams = ContentStream.query.join(User, Network.networkusers).filter(
+            User.id == current_user.id).all()  # created by -> user -> Network
 
     if form.validate_on_submit():
         cleaned_data = form.data  # make a copy
@@ -303,9 +352,13 @@ def music_program_add():
 def music_program_definition(music_program_id):
     music_program = Program.query.filter_by(id=music_program_id).first_or_404()
 
-    # TODO: Filter these by network
-    playlists = ContentMusicPlaylist.query.join(Station).join(Network).join(User, Network.networkusers).filter(User.id == current_user.id).all() #Playlist->Station->Network->user
-    streams = ContentStream.query.join(User, Network.networkusers).filter(User.id == current_user.id).all() # created by -> user -> Network
+    if current_user.role_code == ADMIN:
+        playlists = ContentMusicPlaylist.query.join(Station).join(Network).join(User, Network.networkusers).all() #Playlist->Station->Network->user
+        streams = ContentStream.query.join(User, Network.networkusers).all() # created by -> user -> Network
+    else:
+        # TODO: Filter these by network
+        playlists = ContentMusicPlaylist.query.join(Station).join(Network).join(User, Network.networkusers).filter(User.id == current_user.id).all() #Playlist->Station->Network->user
+        streams = ContentStream.query.join(User, Network.networkusers).filter(User.id == current_user.id).all() # created by -> user -> Network
 
     # render the program structure
     action_names = []
@@ -692,7 +745,7 @@ def station_logs(station_id):
 @login_required
 def schedule():
     # TODO, if user is authorized to view only one station, redirect them there
-    stations = Station.query.join(Network).join(User, Network.networkusers).filter(User.id == current_user.id).all()
+    stations = Station.get_stations(current_user)
     # stations = Station.query.order_by('name').all()
 
     return render_template('radio/schedules.html',
@@ -717,7 +770,10 @@ def schedule_station(station_id):
 
     form = ScheduleProgramForm()
 
-    all_programs = Program.query.join(Program, Network.programs).join(User, Network.networkusers).filter(
+    if current_user.role_code == ADMIN:
+        all_programs = Program.query.join(Program, Network.programs).join(User, Network.networkusers).all()
+    else:
+        all_programs = Program.query.join(Program, Network.programs).join(User, Network.networkusers).filter(
         User.id == current_user.id).all()
     # TODO: filter by language?