Add methods Channel.delete and Video.delete.
This commit is contained in:
		
							parent
							
								
									3f9565b874
								
							
						
					
					
						commit
						226bc03bff
					
				
					 1 changed files with 13 additions and 0 deletions
				
			
		|  | @ -23,6 +23,13 @@ class Channel(Base): | ||||||
|         self.directory = db_row['directory'] |         self.directory = db_row['directory'] | ||||||
|         self.automark = db_row['automark'] or "pending" |         self.automark = db_row['automark'] or "pending" | ||||||
| 
 | 
 | ||||||
|  |     def delete(self, commit=True): | ||||||
|  |         self.ycdldb.sql_delete(table='videos', pairs={'author_id': self.id}) | ||||||
|  |         self.ycdldb.sql_delete(table='channels', pairs={'id': self.id}) | ||||||
|  | 
 | ||||||
|  |         if commit: | ||||||
|  |             self.ycdldb.commit() | ||||||
|  | 
 | ||||||
|     def has_pending(self): |     def has_pending(self): | ||||||
|         query = 'SELECT 1 FROM videos WHERE author_id == ? AND download == "pending" LIMIT 1' |         query = 'SELECT 1 FROM videos WHERE author_id == ? AND download == "pending" LIMIT 1' | ||||||
|         bindings = [self.id] |         bindings = [self.id] | ||||||
|  | @ -93,6 +100,12 @@ class Video(Base): | ||||||
|         except exceptions.NoSuchChannel: |         except exceptions.NoSuchChannel: | ||||||
|             return None |             return None | ||||||
| 
 | 
 | ||||||
|  |     def delete(self, commit=True): | ||||||
|  |         self.ycdldb.sql_delete(table='videos', pairs={'id': self.id}) | ||||||
|  | 
 | ||||||
|  |         if commit: | ||||||
|  |             self.ycdldb.commit() | ||||||
|  | 
 | ||||||
|     def mark_state(self, state, commit=True): |     def mark_state(self, state, commit=True): | ||||||
|         ''' |         ''' | ||||||
|         Mark the video as ignored, pending, or downloaded. |         Mark the video as ignored, pending, or downloaded. | ||||||
|  |  | ||||||
		Loading…
	
		Reference in a new issue