撤消使用imaplib获取的电子邮件的“标记为已读”状态

时间:2013-09-07 23:14:52

标签: python email imaplib

我写了一个python脚本来获取我的所有gmail。我有成千上万的旧电子邮件,其中大约有10,000封未读。

成功获取所有电子邮件后,我发现gmail已将所有已获取的电子邮件标记为“已读”。这对我来说是灾难性的,因为我只需要检查所有未读的电子邮件。

如何恢复哪些电子邮件未读的信息?我将每个邮件对象转储到文件中,我的代码核心如下所示:

m = imaplib.IMAP4_SSL("imap.gmail.com")
m.login(user,pwd)
m.select("[Gmail]/All Mail") 
resp, items = m.uid('search', None, 'ALL')
uids = items[0].split() 
for uid in uids:
    resp, data = m.uid('fetch', uid, "(RFC822)") 
    email_body = data[0][1]
    mail = email.message_from_string(email_body)
    dumbobj(uid, mail)

我希望有一个选项可以在gmail中撤消这个,或者反映所见状态信息的存储邮件对象中的成员。

对于任何希望防止这种头痛的人,请考虑这个答案here。然而,这对我不起作用,因为已经造成了损害。

编辑: 我已经编写了以下函数来递归地“grep”对象中的所有字符串,并使用以下关键字将其应用于转储的电子邮件对象:

regex = "(?i)((marked)|(seen)|(unread)|(read)|(flag)|(delivered)|(status)|(sate))"

到目前为止,没有结果(只有一个无关的“Delivered-To”)。我可以尝试哪些其他关键字?

def grep_object (obj, regex , cycle = set(), matched = set()):
    import re
    if id(obj) in cycle:
        return 
    cycle.update([id(obj)])
    if isinstance(obj, basestring):
        if re.search(regex, obj):
            matched.update([obj])

    def grep_dict (adict ):
        try:
             [  [ grep_object(a, regex, cycle, matched )  for a in ab ] for ab in adict.iteritems() ]
        except:pass

    grep_dict(obj)
    try:grep_dict(obj.__dict__)
    except:pass
    try:
        [ grep_object(elm, regex, cycle, matched ) for elm in obj ]
    except: pass
    return matched

grep_object(mail_object, regex)

1 个答案:

答案 0 :(得分:1)

我有类似的问题(不是用gmail),对我来说最大的问题是制作一个可重复的测试用例;我终于设法制作了一个(见下文)。

Seen标志而言,我现在收集它是这样的:

  • 如果邮件是新的/未显示的,则\Seen标志的IMAP提取将返回空(即,它将不存在,与电子邮件消息相关)。
  • 如果您在邮箱(INBOX)上进行IMAP选择,则会得到一个"标记" UNSEEN,其中包含该文件夹中新的电子邮件ID(或uid)列表(没有\Seen标记)
  • 在我的测试用例中,如果您为BODY.PEEK的邮件提取了标题,那么就不会设置邮件\Seen;如果您使用BODY提取,则设置\Seen
  • 在我的测试用例中,同时提取(RFC822)并未设置\Seen(与Gmail的情况不同)

在测试用例中,我尝试pprint.pprint(inspect.getmembers(mail))(代替您的dumpobj(uid, mail)) - 但仅在我确定\Seen之后。我得到的输出发布在mail_object_inspect.txt - 据我所知,没有提到' new / read / seen'在任何可读字段中都没有;此外mail.as_string()打印:

'From: jesse@example.com\nTo: user@example.com\nSubject: This is a test message!\n\nHello. I am executive assistant to the director of\nBear Stearns, a failed investment Bank.  I have\naccess to USD6,000,000. ...\n'

更糟糕的是,没有提及"字段" imaplib代码中的任何位置(如果文件名不包含不区分大小写的"字段"在任何地方),则打印文件名:

$ grep -L -i field /usr/lib/python{2.7,3.2}/imaplib.py
/usr/lib/python2.7/imaplib.py
/usr/lib/python3.2/imaplib.py

...所以我猜这些信息没有随你的转储一起保存。


这里有一点重构测试用例。最困难的是找到一个小型IMAP服务器,可以使用一些任意用户和电子邮件快速运行,但无需在系统上安装大量内容。最后我找到了一个:trivial-server.pl,Perl' s Net::IMAP::Server的示例文件;在Ubuntu 11.04上测试过。

测试用例粘贴在this gist中,有两个文件(包含很多评论),我会尝试发帖删除:

  • trivial-serverB.pl - Perl(v5.10.1)Net::IMAP::Server服务器(在文件末尾有一个终端输出粘贴,带有telnet客户端会话)
  • testimap.py - Python 2.7 / 3.2 imaplib
    客户端(在文件末尾有一个终端输出粘贴,它本身与服务器一起运行)

trivial-serverB.pl

首先,请确保您有Net::IMAP::Server - 注意,它有很多依赖项,因此以下命令可能需要一段时间才能安装:

sudo perl -MCPAN -e 'install Net::IMAP::Server'

然后,在您获得trivial-serverB.pl的目录中,创建一个包含SSL证书的子目录:

mkdir certs
openssl req \
  -x509 -nodes -days 365 \
  -subj '/C=US/ST=Oregon/L=Portland/CN=localhost' \
  -newkey rsa:1024 -keyout certs/server-key.pem -out certs/server-cert.pem

最后使用管理属性运行服务器:

sudo perl trivial-serverB.pl

请注意,trivial-serverB.pl有一个hack,它允许客户端在没有SSL的情况下进行连接。这是trivial-serverB.pl

#!/usr/bin/perl

use v5.10.1;
use feature qw(say);
use Net::IMAP::Server;

package Demo::IMAP::Hack;
$INC{'Demo/IMAP/Hack.pm'} = 1;

sub capabilityb {
  my $self = shift;
  print STDERR "Capabilitin'\n";
  my $base = $self->server->capability;
  my @words = split " ", $base;
  @words = grep {$_ ne "STARTTLS"} @words
    if $self->is_encrypted;
  unless ($self->auth) {
    my $auth = $self->auth || $self->server->auth_class->new;
    my @auth = $auth->sasl_provides;
    # hack:
    #unless ($self->is_encrypted) {
    #  # Lack of encrpytion makes us turn off all plaintext auth
    #  push @words, "LOGINDISABLED";
    #  @auth = grep {$_ ne "PLAIN"} @auth;
    #}
    push @words, map {"AUTH=$_"} @auth;
  }
  return join(" ", @words);
}

package Demo::IMAP::Auth;
$INC{'Demo/IMAP/Auth.pm'} = 1;
use base 'Net::IMAP::Server::DefaultAuth';
sub auth_plain {
    my ( $self, $user, $pass ) = @_;
    # XXX DO AUTH CHECK
    $self->user($user);
    return 1;
}

package Demo::IMAP::Model;
$INC{'Demo/IMAP/Model.pm'} = 1;
use base 'Net::IMAP::Server::DefaultModel';
sub init {
    my $self = shift;
    $self->root( Demo::IMAP::Mailbox->new() );
    $self->root->add_child( name => "INBOX" );
}

###########################################
package Demo::IMAP::Mailbox;
use base qw/Net::IMAP::Server::Mailbox/;
use Data::Dumper;

my $data = <<'EOF';
From: jesse@example.com
To: user@example.com
Subject: This is a test message!

Hello. I am executive assistant to the director of
Bear Stearns, a failed investment Bank.  I have
access to USD6,000,000. ...
EOF
my $msg = Net::IMAP::Server::Message->new($data);
sub load_data {
    my $self = shift;
    $self->add_message($msg);
}
my %ports = ( port => 143, ssl_port => 993 );
$ports{$_} *= 10 for grep {$> > 0} keys %ports;

$myserv = Net::IMAP::Server->new(
    auth_class  => "Demo::IMAP::Auth",
    model_class => "Demo::IMAP::Model",
    user        => 'nobody',
    log_level   => 3, # at least 3 to output 'CONNECT TCP Peer: ...' message; 4 to output IMAP commands too
    %ports,
);

# apparently, this overload MUST be after the new?! here:
{
no strict 'refs';
*Net::IMAP::Server::Connection::capability = \&Demo::IMAP::Hack::capabilityb;
}

# https://stackoverflow.com/questions/27206371/printing-addresses-of-perl-object-methods
say " -", $myserv->can('validate'), " -", $myserv->can('capability'), " -", \&Net::IMAP::Server::Connection::capability, " -", \&Demo::IMAP::Hack::capabilityb;

$myserv->run();

testimap.py

上面的服务器在一个终端上运行,在另一个终端中你可以这样做:

python testimap.py

代码将简单地从上面的服务器提供的一条(且唯一的)消息中读取字段和内容,并最终恢复(删除)\Seen字段。

import sys
if sys.version_info[0] < 3: # python 2.7
  def uttc(x):
    return x
else:                       # python 3+
  def uttc(x):
    return x.decode("utf-8")
import imaplib
import email
import pprint,inspect

imap_user = 'nobody'
imap_password = 'whatever'
imap_server = 'localhost'
conn = imaplib.IMAP4(imap_server)
conn.debug = 3

try:
  (retcode, capabilities) = conn.login(imap_user, imap_password)
except:
  print(sys.exc_info()[1])
  sys.exit(1)

# not conn.select(readonly=1), else we cannot modify the \Seen flag later
conn.select() # Select inbox or default namespace
(retcode, messages) = conn.search(None, '(UNSEEN)')
if retcode == 'OK':
  for num in uttc(messages[0]).split(' '):
    if not(num):
      print("No messages available: num is `{0}`!".format(num))
      break
    print('Processing message: {0}'.format(num))

    typ, data = conn.fetch(num,'(FLAGS)')
    isSeen = ( "Seen" in uttc(data[0]) )
    print('Got flags: {2}: {0} .. {1}'.format(typ,data, # NEW: OK .. ['1 (FLAGS ())']
            "Seen" if isSeen else "NEW"))

    print('Peeking headers, message: {0} '.format(num))
    typ, data = conn.fetch(num,'(BODY.PEEK[HEADER])')
    pprint.pprint(data)

    typ, data = conn.fetch(num,'(FLAGS)')
    isSeen = ( "Seen" in uttc(data[0]) )
    print('Got flags: {2}: {0} .. {1}'.format(typ,data, # NEW: OK .. ['1 (FLAGS ())']
            "Seen" if isSeen else "NEW"))

    print('Get RFC822 body, message: {0} '.format(num))
    typ, data = conn.fetch(num,'(RFC822)')
    mail = email.message_from_string(uttc(data[0][1]))
    #pprint.pprint(inspect.getmembers(mail))

    typ, data = conn.fetch(num,'(FLAGS)')
    isSeen = ( "Seen" in uttc(data[0]) )
    print('Got flags: {2}: {0} .. {1}'.format(typ,data, # NEW: OK .. ['1 (FLAGS ())']
            "Seen" if isSeen else "NEW"))

    print('Get headers, message: {0} '.format(num))
    typ, data = conn.fetch(num,'(BODY[HEADER])') # note, FLAGS (\\Seen) is now in data, even if not explicitly requested!
    pprint.pprint(data)

    print('Get RFC822 body, message: {0} '.format(num))
    typ, data = conn.fetch(num,'(RFC822)')
    mail = email.message_from_string(uttc(data[0][1]))
    pprint.pprint(inspect.getmembers(mail)) # this is in mail_object_inspect.txt
    pprint.pprint(mail.as_string())

    typ, data = conn.fetch(num,'(FLAGS)')
    isSeen = ( "Seen" in uttc(data[0]) )
    print('Got flags: {2}: {0} .. {1}'.format(typ,data, # Seen: OK .. ['1 (FLAGS (\\Seen))']
            "Seen" if isSeen else "NEW"))

    conn.select() # select again, to see flags server side
    # * OK [UNSEEN 0] # no more unseen messages (if there was only one msg in folder)

    print('Restoring flag to unseen/new, message: {0} '.format(num))
    ret, data = conn.store(num,'-FLAGS','\\Seen')
    if ret == 'OK':
      print("Set back to unseen; Got OK: {0}{1}{2}".format(data,'\n',30*'-'))
      print(mail)

      typ, data = conn.fetch(num,'(FLAGS)')
      isSeen = ( "Seen" in uttc(data[0]) )
      print('Got flags: {2}: {0} .. {1}'.format(typ,data, # NEW: OK .. [b'1 (FLAGS ())']
              "Seen" if isSeen else "NEW"))

conn.close()

参考