annotate pycaptcha/Captcha/File.py @ 148:72a2d431d047

Rajout d'un seed random et d'une fonction get_seed
author SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
date Wed, 24 Feb 2010 13:14:02 -0500
parents 4775b4195b4b
children
rev   line source
87
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
1 """ Captcha.File
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
2
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
3 Utilities for finding and picking random files from our 'data' directory
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
4 """
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
5 #
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
6 # PyCAPTCHA Package
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
7 # Copyright (C) 2004 Micah Dowty <micah@navi.cx>
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
8 #
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
9
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
10 import os, random
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
11
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
12 # Determine the data directory. This can be overridden after import-time if needed.
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
13 dataDir = os.path.join(os.path.split(os.path.abspath(__file__))[0], "data")
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
14
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
15
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
16 class RandomFileFactory(object):
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
17 """Given a list of files and/or directories, this picks a random file.
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
18 Directories are searched for files matching any of a list of extensions.
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
19 Files are relative to our data directory plus a subclass-specified base path.
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
20 """
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
21 extensions = []
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
22 basePath = "."
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
23
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
24 def __init__(self, *fileList):
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
25 self.fileList = fileList
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
26 self._fullPaths = None
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
27
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
28 def _checkExtension(self, name):
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
29 """Check the file against our given list of extensions"""
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
30 for ext in self.extensions:
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
31 if name.endswith(ext):
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
32 return True
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
33 return False
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
34
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
35 def _findFullPaths(self):
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
36 """From our given file list, find a list of full paths to files"""
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
37 paths = []
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
38 for name in self.fileList:
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
39 path = os.path.join(dataDir, self.basePath, name)
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
40 if os.path.isdir(path):
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
41 for content in os.listdir(path):
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
42 if self._checkExtension(content):
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
43 paths.append(os.path.join(path, content))
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
44 else:
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
45 paths.append(path)
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
46 return paths
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
47
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
48 def pick(self):
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
49 if self._fullPaths is None:
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
50 self._fullPaths = self._findFullPaths()
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
51 return random.choice(self._fullPaths)
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
52
4775b4195b4b code pour la generation de captchas
goldfinger
parents:
diff changeset
53 ### The End ###