For my first week post-op I was basically just taking extended naps - get up, do some stuff, go to the bathroom, have a snack, drink some water, back to sleep, repeat 10x or so throughout the day. I really didn't have a bedtime or a wake-up time for that week. I took the ambien around 9pm usually to try to sleep through the night, but my roommate wakes up for work at 3:30, so that's around when I'd wake up and start taking naps.
After drain removal, I went into work the next day at 9 and after that, went back to my 5am wake ups. It was rough, but I just forced myself to do it. If I felt like I needed it, I would take a nap when I got home for about an hour or so.