@@ -30,32 +30,12 @@ class User(UserMixin, db.Model):
3030    da_password_encrypted  =  db .Column (db .Text , nullable = True )
3131    da_domain  =  db .Column (db .String (255 ), nullable = True )
3232
33+     # User preferences 
34+     theme_preference  =  db .Column (db .String (20 ), default = 'light' , nullable = True )
35+ 
3336    # Unique encryption key per user for DA password 
3437    encryption_key  =  db .Column (db .String (255 ), nullable = True )
3538
36-     def  get_theme_preference (self ):
37-         """Get theme preference - uses session storage until database column exists""" 
38-         try :
39-             # Try to get from database column if it exists 
40-             if  hasattr (self , 'theme_preference' ) and  hasattr (self .__class__ , 'theme_preference' ):
41-                 return  getattr (self , 'theme_preference' , 'light' ) or  'light' 
42-         except :
43-             pass 
44-         # Default to light theme 
45-         return  'light' 
46-     
47-     def  set_theme_preference (self , theme ):
48-         """Set theme preference - will work once database column exists""" 
49-         try :
50-             if  theme  in  ['light' , 'dark' ]:
51-                 # Try to set database column if it exists 
52-                 if  hasattr (self .__class__ , 'theme_preference' ):
53-                     setattr (self , 'theme_preference' , theme )
54-                     return  True 
55-         except :
56-             pass 
57-         return  False 
58- 
5939    def  __init__ (self , ** kwargs ):
6040        """Initialize user with encryption key""" 
6141        super (User , self ).__init__ (** kwargs )
0 commit comments