Changed import to not import entire datetime library.
This commit is contained in:
		
							
								
								
									
										50
									
								
								torrentSearch/utils.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										50
									
								
								torrentSearch/utils.py
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,50 @@ | |||||||
|  | #!/usr/bin/env python3.6 | ||||||
|  | # -*- coding: utf-8 -*- | ||||||
|  | # @Author: KevinMidboe | ||||||
|  | # @Date:   2017-11-01 15:57:23 | ||||||
|  | # @Last Modified by:   KevinMidboe | ||||||
|  | # @Last Modified time: 2017-11-02 15:14:31 | ||||||
|  |  | ||||||
|  | import re | ||||||
|  | from datetime import datetime | ||||||
|  |  | ||||||
|  | def sanitize(string, ignore_characters=None, replace_characters=None): | ||||||
|  | 	"""Sanitize a string to strip special characters. | ||||||
|  |  | ||||||
|  | 	:param str string: the string to sanitize. | ||||||
|  | 	:param set ignore_characters: characters to ignore. | ||||||
|  | 	:return: the sanitized string. | ||||||
|  | 	:rtype: str | ||||||
|  |  | ||||||
|  | 	""" | ||||||
|  | 	# only deal with strings | ||||||
|  | 	if string is None: | ||||||
|  | 		return | ||||||
|  | 	 | ||||||
|  | 	replace_characters = replace_characters or '' | ||||||
|  |  | ||||||
|  | 	ignore_characters = ignore_characters or set() | ||||||
|  |  | ||||||
|  | 	characters = ignore_characters | ||||||
|  | 	if characters: | ||||||
|  | 		string = re.sub(r'[%s]' % re.escape(''.join(characters)), replace_characters, string) | ||||||
|  |  | ||||||
|  | 	return string | ||||||
|  |  | ||||||
|  | def return_re_match(string, re_statement): | ||||||
|  | 	if string is None: | ||||||
|  | 		return | ||||||
|  |  | ||||||
|  | 	m = re.search(re_statement, string) | ||||||
|  | 	if 'Y-day' in m.group(): | ||||||
|  | 		return datetime.now().strftime('%m-%d %Y') | ||||||
|  | 	return sanitize(m.group(), '\xa0', ' ') | ||||||
|  |  | ||||||
|  |  | ||||||
|  | # Can maybe be moved away from this class | ||||||
|  | # returns a number that is either the value of multiple_pages | ||||||
|  | # or if it exceeds total_pages, return total_pages. | ||||||
|  | def pagesToCount(multiple, total): | ||||||
|  | 	if (multiple > total): | ||||||
|  | 		return total | ||||||
|  | 	return multiple | ||||||
		Reference in New Issue
	
	Block a user