kallithea: rhodecode/lib/__init_

comparison rhodecode/lib/init.py @ 2031:82a88013a3fd

merge 1.3 into stable

author	Marcin Kuzminski <marcin@python-works.com>
date	Sun, 26 Feb 2012 17:25:09 +0200
parents	54e84659cb3a 6020e3884a58
children	abe75448253c

comparison

equal deleted inserted replaced

-:ab0e122b38a7
+:82a88013a3fd
 #
 # You should have received a copy of the GNU General Public License
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 import os
+import re
+from rhodecode.lib.vcs.utils.lazy import LazyProperty
 def __get_lem():
 from pygments import lexers
 from string import lower
 from collections import defaultdict
 # NOTE: that this will overide any mappings in LANGUAGES_EXTENSIONS_MAP
 ADDITIONAL_MAPPINGS = {'xaml': 'XAML'}
 LANGUAGES_EXTENSIONS_MAP.update(ADDITIONAL_MAPPINGS)
+# list of readme files to search in file tree and display in summary
+# attached weights defines the search  order lower is first
+ALL_READMES = [
+('readme', 0), ('README', 0), ('Readme', 0),
+('doc/readme', 1), ('doc/README', 1), ('doc/Readme', 1),
+('Docs/readme', 2), ('Docs/README', 2), ('Docs/Readme', 2),
+('DOCS/readme', 2), ('DOCS/README', 2), ('DOCS/Readme', 2),
+('docs/readme', 2), ('docs/README', 2), ('docs/Readme', 2),
+]
+# extension together with weights to search lower is first
+RST_EXTS = [
+('', 0), ('.rst', 1), ('.rest', 1),
+('.RST', 2), ('.REST', 2),
+('.txt', 3), ('.TXT', 3)
+]
+MARKDOWN_EXTS = [
+('.md', 1), ('.MD', 1),
+('.mkdn', 2), ('.MKDN', 2),
+('.mdown', 3), ('.MDOWN', 3),
+('.markdown', 4), ('.MARKDOWN', 4)
+]
+PLAIN_EXTS = [('.text', 2), ('.TEXT', 2)]
+ALL_EXTS = MARKDOWN_EXTS + RST_EXTS + PLAIN_EXTS
 def str2bool(_str):
 """
 returs True/False value from given string, it tries to translate the
 string into boolean
 line = replace(line, '\r', '\n')
 elif mode == 1:
 line = replace(line, '\r\n', '\r')
 line = replace(line, '\n', '\r')
 elif mode == 2:
-import re
 line = re.sub("\r(?!\n)|(?<!\r)\n", "\r\n", line)
 return line
 def detect_mode(line, default):
 salt = _RandomNameSequence().next()
 return hashlib.sha1(username + salt).hexdigest()
-def safe_unicode(str_, from_encoding='utf8'):
+def safe_unicode(str_, from_encoding=None):
 """
 safe unicode function. Does few trick to turn str_ into unicode
 In case of UnicodeDecode error we try to return it with encoding detected
 by chardet library if it fails fallback to unicode with errors replaced
 :rtype: unicode
 :returns: unicode object
 """
 if isinstance(str_, unicode):
 return str_
+if not from_encoding:
+import rhodecode
+DEFAULT_ENCODING = rhodecode.CONFIG.get('default_encoding','utf8')
+from_encoding = DEFAULT_ENCODING
 try:
 return unicode(str_)
 except UnicodeDecodeError:
 pass
 raise Exception()
 return str_.decode(encoding)
 except (ImportError, UnicodeDecodeError, Exception):
 return unicode(str_, from_encoding, 'replace')
-def safe_str(unicode_, to_encoding='utf8'):
+def safe_str(unicode_, to_encoding=None):
 """
 safe str function. Does few trick to turn unicode_ into string
 In case of UnicodeEncodeError we try to return it with encoding detected
 by chardet library if it fails fallback to string with errors replaced
 :param unicode_: unicode to encode
 :rtype: str
 """
 if not isinstance(unicode_, basestring):
 return str(unicode_)
+if not isinstance(unicode_, basestring):
+return str(unicode_)
 if isinstance(unicode_, str):
 return unicode_
+if not to_encoding:
+import rhodecode
+DEFAULT_ENCODING = rhodecode.CONFIG.get('default_encoding','utf8')
+to_encoding = DEFAULT_ENCODING
 try:
 return unicode_.encode(to_encoding)
 except UnicodeEncodeError:
 pass
 return unicode_.encode(to_encoding, 'replace')
 return safe_str
 def engine_from_config(configuration, prefix='sqlalchemy.', **kwargs):
 """
 Custom engine_from_config functions that makes sure we use NullPool for
 file based sqlite databases. This prevents errors on sqlite. This only
 applies to sqlalchemy versions < 0.7.0
 """
 import sqlalchemy
 from sqlalchemy import engine_from_config as efc
 def age(curdate):
 """
 turns a datetime into an age string.
 :param curdate: datetime object
 :rtype: unicode
 :returns: unicode words describing age
 """
 from datetime import datetime
 from webhelpers.date import time_ago_in_words
-_ = lambda s:s
+_ = lambda s: s
 if not curdate:
 return ''
 agescales = [(_(u"year"), 3600 * 24 * 365),
 age = datetime.now() - curdate
 age_seconds = (age.days * agescales[2][1]) + age.seconds
 pos = 1
 for scale in agescales:
 if scale[1] <= age_seconds:
-if pos == 6:pos = 5
+if pos == 6:
+pos = 5
 return '%s %s' % (time_ago_in_words(curdate,
 agescales[pos][0]), _('ago'))
 pos += 1
 return _(u'just now')
 def uri_filter(uri):
 """
 Removes user:password from given url string
 :param uri:
 :rtype: unicode
 :returns: filtered list of strings
 """
 if not uri:
 return ''
 proto = ''
 def credentials_filter(uri):
 """
 Returns a url with removed credentials
 :param uri:
 """
 uri = uri_filter(uri)
 #check if we have port
 if len(uri) > 2 and uri[2]:
 uri[2] = ':' + uri[2]
 return ''.join(uri)
 def get_changeset_safe(repo, rev):
 """
 Safe version of get_changeset if this changeset doesn't exists for a
 repo it returns a Dummy one instead
 :param repo:
 :param rev:
 """
-from vcs.backends.base import BaseRepository
+from rhodecode.lib.vcs.backends.base import BaseRepository
-from vcs.exceptions import RepositoryError
+from rhodecode.lib.vcs.exceptions import RepositoryError
 if not isinstance(repo, BaseRepository):
 raise Exception('You must pass an Repository '
 'object as first argument got %s', type(repo))
 try:
 def get_current_revision(quiet=False):
 """
 Returns tuple of (number, id) from repository containing this package
 or None if repository could not be found.
 :param quiet: prints error for fetching revision if True
 """
 try:
-from vcs import get_repo
+from rhodecode.lib.vcs import get_repo
-from vcs.utils.helpers import get_scm
+from rhodecode.lib.vcs.utils.helpers import get_scm
 repopath = os.path.join(os.path.dirname(__file__), '..', '..')
 scm = get_scm(repopath)[0]
 repo = get_repo(path=repopath, alias=scm)
 tip = repo.get_changeset()
 return (tip.revision, tip.short_id)
 if not quiet:
 print ("Cannot retrieve rhodecode's revision. Original error "
 "was: %s" % err)
 return None
+def extract_mentioned_users(s):
+"""
+Returns unique usernames from given string s that have @mention
+:param s: string to get mentions
+"""
+usrs = {}
+for username in re.findall(r'(?:^@|\s@)(\w+)', s):
+usrs[username] = username
+return sorted(usrs.keys())

Mercurial > kallithea

comparison rhodecode/lib/__init__.py @ 2031:82a88013a3fd

comparison rhodecode/lib/init.py @ 2031:82a88013a3fd