| 1 |  |  | from plugin.core.helpers.variable import to_integer | 
            
                                                                                                            
                            
            
                                    
            
            
                | 2 |  |  | from plugin.managers.action import ActionManager | 
                            
                    |  |  |  | 
                                                                                        
                                                                                            
                                                                                     | 
            
                                                                                                            
                            
            
                                    
            
            
                | 3 |  |  | from plugin.managers.session.s_websocket import WSessionManager | 
            
                                                                                                            
                            
            
                                    
            
            
                | 4 |  |  | from plugin.scrobbler.core.constants import IGNORED_EVENTS | 
            
                                                                                                            
                            
            
                                    
            
            
                | 5 |  |  | from plugin.scrobbler.core.engine import SessionEngine | 
            
                                                                                                            
                            
            
                                    
            
            
                | 6 |  |  | from plugin.scrobbler.methods.core.base import Base | 
            
                                                                                                            
                            
            
                                    
            
            
                | 7 |  |  |  | 
            
                                                                                                            
                            
            
                                    
            
            
                | 8 |  |  | from datetime import datetime, timedelta | 
            
                                                                                                            
                            
            
                                    
            
            
                | 9 |  |  | from plex import Plex | 
            
                                                                                                            
                            
            
                                    
            
            
                | 10 |  |  | from plex_activity import Activity | 
            
                                                                                                            
                            
            
                                    
            
            
                | 11 |  |  | import logging | 
            
                                                                                                            
                            
            
                                    
            
            
                | 12 |  |  |  | 
            
                                                                                                            
                            
            
                                    
            
            
                | 13 |  |  | log = logging.getLogger(__name__) | 
            
                                                                                                            
                            
            
                                    
            
            
                | 14 |  |  |  | 
            
                                                                                                            
                            
            
                                    
            
            
                | 15 |  |  |  | 
            
                                                                                                            
                            
            
                                    
            
            
                | 16 |  |  | class WebSocket(Base): | 
            
                                                                                                            
                            
            
                                    
            
            
                | 17 |  |  |     name = 'websocket' | 
            
                                                                                                            
                            
            
                                    
            
            
                | 18 |  |  |  | 
            
                                                                                                            
                            
            
                                    
            
            
                | 19 |  |  |     def __init__(self): | 
            
                                                                                                            
                            
            
                                    
            
            
                | 20 |  |  |         Activity.on('websocket.playing', self.on_playing) | 
            
                                                                                                            
                            
            
                                    
            
            
                | 21 |  |  |  | 
            
                                                                                                            
                            
            
                                    
            
            
                | 22 |  |  |         self.engine = SessionEngine() | 
            
                                                                                                            
                            
            
                                    
            
            
                | 23 |  |  |  | 
            
                                                                                                            
                            
            
                                                                    
                                                                                                        
            
            
                | 24 |  | View Code Duplication |     def on_playing(self, info): | 
                            
                    |  |  |  | 
                                                                                        
                                                                                     | 
            
                                                                                                            
                            
            
                                    
            
            
                | 25 |  |  |         # Create or retrieve existing session | 
            
                                                                                                            
                            
            
                                    
            
            
                | 26 |  |  |         session = WSessionManager.get.or_create(info, fetch=True) | 
                            
                    |  |  |  | 
                                                                                        
                                                                                     | 
            
                                                                                                            
                            
            
                                    
            
            
                | 27 |  |  |  | 
            
                                                                                                            
                            
            
                                    
            
            
                | 28 |  |  |         # Validate session | 
            
                                                                                                            
                            
            
                                    
            
            
                | 29 |  |  |         if session.updated_at is None or (datetime.utcnow() - session.updated_at) > timedelta(minutes=5): | 
            
                                                                                                            
                            
            
                                    
            
            
                | 30 |  |  |             log.info('Updating session, last update was over 5 minutes ago') | 
            
                                                                                                            
                            
            
                                    
            
            
                | 31 |  |  |             WSessionManager.update(session, info, fetch=True) | 
            
                                                                                                            
                            
            
                                    
            
            
                | 32 |  |  |             return | 
            
                                                                                                            
                            
            
                                    
            
            
                | 33 |  |  |  | 
            
                                                                                                            
                            
            
                                    
            
            
                | 34 |  |  |         if session.duration is None or session.view_offset is None: | 
            
                                                                                                            
                            
            
                                    
            
            
                | 35 |  |  |             # Update session | 
            
                                                                                                            
                            
            
                                    
            
            
                | 36 |  |  |             WSessionManager.update(session, info, fetch=lambda s, i: ( | 
            
                                                                                                            
                            
            
                                    
            
            
                | 37 |  |  |                 s.rating_key != to_integer(i.get('ratingKey')) or | 
            
                                                                                                            
                            
            
                                    
            
            
                | 38 |  |  |                 s.duration is None | 
            
                                                                                                            
                            
            
                                    
            
            
                | 39 |  |  |             )) | 
            
                                                                                                            
                            
            
                                    
            
            
                | 40 |  |  |             return | 
            
                                                                                                            
                            
            
                                    
            
            
                | 41 |  |  |  | 
            
                                                                                                            
                            
            
                                    
            
            
                | 42 |  |  |         # Parse `info` to events | 
            
                                                                                                            
                            
            
                                    
            
            
                | 43 |  |  |         events = self.to_events(session, info) | 
            
                                                                                                            
                            
            
                                    
            
            
                | 44 |  |  |  | 
            
                                                                                                            
                            
            
                                    
            
            
                | 45 |  |  |         if not events: | 
            
                                                                                                            
                            
            
                                    
            
            
                | 46 |  |  |             return | 
            
                                                                                                            
                            
            
                                    
            
            
                | 47 |  |  |  | 
            
                                                                                                            
                            
            
                                    
            
            
                | 48 |  |  |         # Check for changed media | 
            
                                                                                                            
                            
            
                                    
            
            
                | 49 |  |  |         media_changed = session.rating_key != to_integer(info.get('ratingKey')) | 
            
                                                                                                            
                            
            
                                    
            
            
                | 50 |  |  |  | 
            
                                                                                                            
                            
            
                                    
            
            
                | 51 |  |  |         # Parse `events` | 
            
                                                                                                            
                            
            
                                    
            
            
                | 52 |  |  |         actions = self.engine.process(session, events) | 
            
                                                                                                            
                            
            
                                    
            
            
                | 53 |  |  |  | 
            
                                                                                                            
                            
            
                                    
            
            
                | 54 |  |  |         for action, payload in actions: | 
            
                                                                                                            
                            
            
                                    
            
            
                | 55 |  |  |             # Build request for the event | 
            
                                                                                                            
                            
            
                                    
            
            
                | 56 |  |  |             request = self.build_request( | 
            
                                                                                                            
                            
            
                                    
            
            
                | 57 |  |  |                 session, | 
            
                                                                                                            
                            
            
                                    
            
            
                | 58 |  |  |                 rating_key=payload.get('rating_key'), | 
            
                                                                                                            
                            
            
                                    
            
            
                | 59 |  |  |                 view_offset=payload.get('view_offset') | 
            
                                                                                                            
                            
            
                                    
            
            
                | 60 |  |  |             ) | 
            
                                                                                                            
                            
            
                                    
            
            
                | 61 |  |  |  | 
            
                                                                                                            
                            
            
                                    
            
            
                | 62 |  |  |             if not request: | 
            
                                                                                                            
                            
            
                                    
            
            
                | 63 |  |  |                 continue | 
            
                                                                                                            
                            
            
                                    
            
            
                | 64 |  |  |  | 
            
                                                                                                            
                            
            
                                    
            
            
                | 65 |  |  |             # Queue request to be sent | 
            
                                                                                                            
                            
            
                                    
            
            
                | 66 |  |  |             ActionManager.queue('/'.join(['scrobble', action]), request, session) | 
            
                                                                                                            
                            
            
                                    
            
            
                | 67 |  |  |  | 
            
                                                                                                            
                            
            
                                    
            
            
                | 68 |  |  |         # Update session | 
            
                                                                                                            
                            
            
                                    
            
            
                | 69 |  |  |         WSessionManager.update(session, info, fetch=media_changed) | 
            
                                                                                                            
                                                                
            
                                    
            
            
                | 70 |  |  |  | 
            
                                                                        
                            
            
                                                                    
                                                                                                        
            
            
                | 71 |  | View Code Duplication |     @classmethod | 
                            
                    |  |  |  | 
                                                                                        
                                                                                     | 
            
                                                                        
                            
            
                                    
            
            
                | 72 |  |  |     def to_events(cls, session, info): | 
            
                                                                        
                            
            
                                    
            
            
                | 73 |  |  |         # Validate `state` | 
            
                                                                        
                            
            
                                    
            
            
                | 74 |  |  |         state = info.get('state') | 
            
                                                                        
                            
            
                                    
            
            
                | 75 |  |  |  | 
            
                                                                        
                            
            
                                    
            
            
                | 76 |  |  |         if not state: | 
            
                                                                        
                            
            
                                    
            
            
                | 77 |  |  |             log.warn('Event has an invalid state %r', state) | 
            
                                                                        
                            
            
                                    
            
            
                | 78 |  |  |             return [] | 
            
                                                                        
                            
            
                                    
            
            
                | 79 |  |  |  | 
            
                                                                        
                            
            
                                    
            
            
                | 80 |  |  |         if state in IGNORED_EVENTS: | 
            
                                                                        
                            
            
                                    
            
            
                | 81 |  |  |             log.debug('Ignored "%s" event: %r', state, info) | 
            
                                                                        
                            
            
                                    
            
            
                | 82 |  |  |             return [] | 
            
                                                                        
                            
            
                                    
            
            
                | 83 |  |  |  | 
            
                                                                        
                            
            
                                    
            
            
                | 84 |  |  |         # Check for session `view_offset` jump | 
            
                                                                        
                            
            
                                    
            
            
                | 85 |  |  |         if cls.session_jumped(session, info.get('viewOffset')): | 
            
                                                                        
                            
            
                                    
            
            
                | 86 |  |  |             return [] | 
            
                                                                        
                            
            
                                    
            
            
                | 87 |  |  |  | 
            
                                                                        
                            
            
                                    
            
            
                | 88 |  |  |         # Build event | 
            
                                                                        
                            
            
                                    
            
            
                | 89 |  |  |         return [ | 
            
                                                                        
                            
            
                                    
            
            
                | 90 |  |  |             (state, { | 
            
                                                                        
                            
            
                                    
            
            
                | 91 |  |  |                 'rating_key': to_integer(info.get('ratingKey')), | 
            
                                                                        
                            
            
                                    
            
            
                | 92 |  |  |                 'view_offset': to_integer(info.get('viewOffset')) | 
            
                                                                                                            
                            
            
                                    
            
            
                | 93 |  |  |             }) | 
            
                                                                                                            
                            
            
                                    
            
            
                | 94 |  |  |         ] | 
            
                                                                                                            
                            
            
                                    
            
            
                | 95 |  |  |  | 
            
                                                                                                            
                            
            
                                    
            
            
                | 96 |  |  |     @classmethod | 
            
                                                                                                            
                            
            
                                    
            
            
                | 97 |  |  |     def test(cls): | 
            
                                                                                                            
                            
            
                                    
            
            
                | 98 |  |  |         if Plex['status'].sessions() is None: | 
            
                                                                                                            
                            
            
                                    
            
            
                | 99 |  |  |             log.info("Error while retrieving sessions, assuming WebSocket method isn't available") | 
            
                                                                                                            
                            
            
                                    
            
            
                | 100 |  |  |             return False | 
            
                                                                                                            
                            
            
                                    
            
            
                | 101 |  |  |  | 
            
                                                                                                            
                            
            
                                    
            
            
                | 102 |  |  |         detail = Plex.detail() | 
                            
                    |  |  |  | 
                                                                                        
                                                                                     | 
            
                                                                                                            
                            
            
                                    
            
            
                | 103 |  |  |         if detail is None: | 
            
                                                                                                            
                            
            
                                    
            
            
                | 104 |  |  |             log.info('Error while retrieving server info for testing') | 
            
                                                                                                            
                            
            
                                    
            
            
                | 105 |  |  |             return False | 
            
                                                                                                            
                            
            
                                    
            
            
                | 106 |  |  |  | 
            
                                                                                                            
                            
            
                                    
            
            
                | 107 |  |  |         if not detail.multiuser: | 
            
                                                                                                            
                            
            
                                    
            
            
                | 108 |  |  |             log.info("Server info indicates multi-user support isn't available, WebSocket method not available") | 
            
                                                                                                            
                            
            
                                    
            
            
                | 109 |  |  |             return False | 
            
                                                                                                            
                            
            
                                    
            
            
                | 110 |  |  |  | 
            
                                                                                                            
                                                                
            
                                    
            
            
                | 111 |  |  |         return True | 
            
                                                        
            
                                    
            
            
                | 112 |  |  |  |